Trait safetensors::slice::IndexOp

pub trait IndexOp<'data, T> {
    // Required method
    fn slice(
        &'data self,
        index: T
    ) -> Result<SliceIterator<'data>, InvalidSlice>;
}

Trait used to implement multiple signatures for ease of use of the slicing of a tensor

Required Methods

fn slice(&'data self, index: T) -> Result<SliceIterator<'data>, InvalidSlice>

Returns a slicing iterator which are the chunks of data necessary to reconstruct the desired tensor.

Implementors

impl<'data, A> IndexOp<'data, (A,)> for TensorView<'data>

where A: Into<TensorIndexer>,

impl<'data, A> IndexOp<'data, A> for TensorView<'data>

where A: Into<TensorIndexer>,

impl<'data, A, B> IndexOp<'data, (A, B)> for TensorView<'data>

where A: Into<TensorIndexer>, B: Into<TensorIndexer>,

impl<'data, A, B, C> IndexOp<'data, (A, B, C)> for TensorView<'data>

where A: Into<TensorIndexer>, B: Into<TensorIndexer>, C: Into<TensorIndexer>,