def slice_dim

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

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

1466def slice_dim(+slicing: Slicing) -> Signed.Position:
1467  match slicing:
1468    case Sliced{dim, start, stop, step}:
1469      dim
1470    case Narrowed{dim, start, length}:
1471      dim
1472    case Selected{dim, index}:
1473      dim

Uses (2)

Used by (1)