TensorDimensionIndexSelection

Indexing a specific tensor dimension.

Selecting dimension=2 and index=42 is similar to doing tensor[:, :, 42, :, :, …] in numpy.

Fields fields

Type: uint32

The dimension number to select.

Type: uint64

The index along the dimension to use.

Struct {
    dimension: uint32
    index: uint64
}