def take

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

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

253def take(index: Array<U32>, pair: Array<F32> & F32) -> F32 & (Array<F32> & Array<U32>):
254  (a, x) = pair
255  (x, (a, index))

Uses (5)

Used by (5)