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

Used by used-by

• TensorDimensionIndexSelection