Enum ArrayIndexOp

pub enum ArrayIndexOp<'a> {
    Ellipsis,
    TakeIndex {
        index: i32,
    },
    TakeArray {
        indices: Rc<Array>,
    },
    TakeArrayRef {
        indices: &'a Array,
    },
    Slice(RangeIndex),
    ExpandDims,
}

Indexing operation for arrays.

Variants

Ellipsis

An Ellipsis is used to consume all axes

This is equivalent to ... in python

TakeIndex

A single index operation

This is equivalent to arr[1] in python

Fields

index: i32

The index to take

TakeArray

Indexing with an array

Fields

indices: Rc<Array>

The indices to take

TakeArrayRef

Indexing with an array reference

Fields

indices: &'a Array

The indices to take

Slice(RangeIndex)

Indexing with a range

This is equivalent to arr[1:3] in python

ExpandDims

New axis operation

This is equivalent to arr[None] in python

Trait Implementations

impl<'a> ArrayIndex<'a> for ArrayIndexOp<'a>

fn index_op(self) -> ArrayIndexOp<'a>

mlx allows out of bounds indexing.

impl<'a> Clone for ArrayIndexOp<'a>

fn clone(&self) -> ArrayIndexOp<'a>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<'a> Debug for ArrayIndexOp<'a>

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

Formats the value using the given formatter.