Struct TensorSlice 

pub struct TensorSlice { /* private fields */ }

A validated contiguous slice of a source tensor.

Implementations

impl TensorSlice

pub fn for_shape( shape: &[usize], axis: usize, index: usize, parts: usize, ) -> Result<Self, PlacementPlanError>

Validates and calculates an equal contiguous tensor slice.

pub const fn axis(&self) -> usize

Returns the source axis.

pub const fn start(&self) -> usize

Returns the inclusive source offset.

pub const fn end(&self) -> usize

Returns the exclusive source offset.

pub const fn index(&self) -> usize

Returns the shard index.

pub const fn parts(&self) -> usize

Returns the shard count.

pub fn local_shape(&self, source_shape: &[usize]) -> Vec<usize>

Returns the resulting local shape.

Trait Implementations

impl Clone for TensorSlice

fn clone(&self) -> TensorSlice

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorSlice

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

Formats the value using the given formatter.

impl Eq for TensorSlice

impl PartialEq for TensorSlice

fn eq(&self, other: &TensorSlice) -> bool

Equality operator ==.

fn ne(&self, other: &Rhs) -> bool

Inequality operator !=.

impl StructuralPartialEq for TensorSlice