type Indexing

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

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

1535# Which of the two an indexing step is: PyTorch's a.index_select(dim, index)
1536# with a vector of entries, and a.gather(dim, index) with an index tensor.
1537type Indexing is Data:
1538  IndexSelected{dim: Signed.Position, entries: List<&2, U32>}
1539  GatheredBy{dim: Signed.Position, index: Gather.Index}