def gather

bendtorch-core@0.1.0.0 · src/autograd/tensor_backward.bend:1605

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

1605# Records a.gather(dim, index).
1606def gather(tape: Tape, +a: Nat, +dim: Signed.Position, +index: Gather.Index) -> Tape & Result<&2, &2, Error, Nat>:
1607  indexed(tape, a, GatheredBy{dim, index})