Enum TensorIndexer

pub enum TensorIndexer {
    Select(usize),
    Narrow(Bound<usize>, Bound<usize>),
    IndexSelect(Tensor),
    Err(Error),
}

Generic structure used to index a slice of the tensor

Variants

Select(usize)

This selects the elements for which an index has some specific value.

Narrow(Bound<usize>, Bound<usize>)

This is a regular slice, purely indexing a chunk of the tensor

IndexSelect(Tensor)

Indexing via a 1d tensor

Err(Error)

Trait Implementations

impl Debug for TensorIndexer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl From<&[u32]> for TensorIndexer

fn from(index: &[u32]) -> Self

Converts to this type from the input type.

impl From<&Tensor> for TensorIndexer

fn from(tensor: &Tensor) -> Self

Converts to this type from the input type.

impl<T: RB> From<T> for TensorIndexer

fn from(range: T) -> Self

Converts to this type from the input type.

impl From<Vec<u32>> for TensorIndexer

fn from(index: Vec<u32>) -> Self

Converts to this type from the input type.

impl From<usize> for TensorIndexer

fn from(index: usize) -> Self

Converts to this type from the input type.