Enum TensorIndex 

pub enum TensorIndex {
    Index(i64),
    Range(Option<i64>, Option<i64>, Option<i64>),
    All,
    List(Vec<i64>),
    Mask(Tensor<bool>),
    Ellipsis,
    NewAxis,
}

Index type for tensor indexing

Variants

Index(i64)

Single index

Range(Option<i64>, Option<i64>, Option<i64>)

Range of indices

All

All indices (:)

List(Vec<i64>)

List of indices (fancy indexing)

Mask(Tensor<bool>)

Boolean mask

Ellipsis

Ellipsis (…) - represents multiple ‘:’ to fill remaining dimensions

NewAxis

Newaxis (None) - adds a dimension of size 1

Implementations

impl TensorIndex

pub fn range(start: Option<i64>, stop: Option<i64>) -> Self

Create a range index

pub fn range_step(start: Option<i64>, stop: Option<i64>, step: i64) -> Self

Create a range index with step

Trait Implementations

impl Clone for TensorIndex

fn clone(&self) -> TensorIndex

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TensorIndex

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

Formats the value using the given formatter.