def looked

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

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

import Base

257def looked(+plan: Plan, +at: Nat, a: Array<F32>, pair: Array<U32> & U32) -> F32 & (Array<F32> & Array<U32>):
258  (index, +j) = pair
259  take(index, Array.get(F32, a, U32.from_nat(source(plan, at, U32.to_nat(j)))))

Uses (12)

Used by (5)