TensorDimensionIndexSelection
Indexing a specific tensor dimension.
Selecting dimension=2
and index=42
is similar to doing tensor[:, :, 42, :, :, …]
in numpy.
Fields fields
- dimension:
u32
- index:
u64
API reference links api-reference-links
- 🌊 C++ API docs for
TensorDimensionIndexSelection
- 🐍 Python API docs for
TensorDimensionIndexSelection
- 🦀 Rust API docs for
TensorDimensionIndexSelection