def scatter

bendtorch-core@0.1.0.0 · src/tensor/axis_spec.bend:56

import Base

56def scatter(-A: Data, rank: Nat, +axes: List<&2, Nat>, +coordinates: List<&2, A>, +cursor: Nat) -> Maybe<&2, List<&2, A>>:
57  match rank:
58    case 0n:
59      Some{Nil{}}
60    case 1n+p:
61      prepend(A, select(A, coordinates, position(axes, cursor)), scatter(A, p, axes, coordinates, 1n+cursor))