def compared

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

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

304def compared(+i: Nat, +at: Nat, acc: F32, g: Array<F32>, pair: Array<U32> & U32) -> F32 & (Array<F32> & Array<U32>):
305  (index, +j) = pair
306  summed(Nat.is_eq(U32.to_nat(j), i), at, acc, g, index)

Uses (9)

Used by (3)