def take
import bendtorch-core@0.1.0.0/src/tensor/gather.bend as Gatherimport Base
253def take(index: Array<U32>, pair: Array<F32> & F32) -> F32 & (Array<F32> & Array<U32>):
254 (a, x) = pair
255 (x, (a, index))
import bendtorch-core@0.1.0.0/src/tensor/gather.bend as Gatherimport Base
253def take(index: Array<U32>, pair: Array<F32> & F32) -> F32 & (Array<F32> & Array<U32>):
254 (a, x) = pair
255 (x, (a, index))