def index_select

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

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

1601# Records a.index_select(dim, index).
1602def index_select(tape: Tape, +a: Nat, +dim: Signed.Position, +entries: List<&2, U32>) -> Tape & Result<&2, &2, Error, Nat>:
1603  indexed(tape, a, IndexSelected{dim, entries})