TensorDimensionIndexSelection

Indexing a specific tensor dimension.

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

Fields fields

dimension

Type: non-null UInt32

The dimension number to select.

index

Type: non-null UInt64

The index along the dimension to use.

Arrow datatype arrow-datatype

Struct(
    "dimension": non-null UInt32
    "index": non-null UInt64
)

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