def select

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

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

1523# Records a.select(dim, index), which drops the axis.
1524def select(tape: Tape, +a: Nat, +dim: Signed.Position, +index: Signed.Position) -> Tape & Result<&2, &2, Error, Nat>:
1525  sliced(tape, a, Selected{dim, index})