type Index

bendtorch-core@0.1.0.0 · src/tensor/gather.bend:48

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

48# PyTorch's index tensor for gather: its shape, a Shape as every tensor's
49# (rank at most 4, at most 2^20 elements), and its entries in row-major
50# order.
51type Index is Data:
52  Index{shape: Shape.Shape, entries: List<&2, U32>}