def scattered

bendtorch-core@0.1.0.0 · src/tensor/gather.bend:336

import bendtorch-core@0.1.0.0/src/tensor/gather.bend as Gather

import Base

336# Input cell `at` is (o, i, r) of [outer, extent, inner].
337def scattered(+plan: Plan, +at: Nat, state: Array<F32> & Array<U32>) -> F32 & (Array<F32> & Array<U32>):
338  Plan{+kind, +outer, +extent, +inner, +length} = plan
339  +row = Nat.div(at, inner)
340  added(plan, Nat.div(row, extent), Nat.mod(row, extent), Nat.mod(at, inner), length, 0n, (0.0, state))

Uses (10)

Used by (3)