def slice

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

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

1514# Records a[..., start:stop:step, ...] with the slice at axis `dim`.
1515def slice(tape: Tape, +a: Nat, +dim: Signed.Position, +start: Signed.Position, +stop: Signed.Position, +step: U32) ->
1516  Tape & Result<&2, &2, Error, Nat>:
1517  sliced(tape, a, Sliced{dim, start, stop, step})