Struct tension::Shape

pub struct Shape { /* fields omitted */ }

Structure representing shape of the tensor.

Tensor supposed to have an infinite number of axes. E.g. tensor of shape (x,y,z) supposed to be (x,y,z,1,1,1,...). That means that trailing axes of size 1 are ignored, so shapes (x,y,z) and (x,y,z,1) are equal.

There may be a tensor of shape (,) (0-dimensional tensor or scalar).

Implementations

impl Shape

pub fn len(&self) -> usize

Count of dimensions without trailing 1s in the end.

pub fn as_slice(&self) -> &[usize]

Slice of dimension sizes.

pub fn as_mut_slice(&mut self) -> &mut [usize]

Mutable slice of dimension sizes.

pub fn iter(&self) -> Iter<'_, usize>

Iterator over dimension sizes.

pub fn iter_mut(&mut self) -> IterMut<'_, usize>

Mutable iterator over dimension sizes.

pub fn content(&self) -> usize

Number of items in tensor of this shape.

impl Shape

pub fn slice<R: RangeBounds<usize>>(&self, range: R) -> Shape

Slicing operation.

Returns a new shape sliced from the original one.

Sadly, you cannot use shape[a..b] syntax because Rust Index trait is required to return a reference.

Trait Implementations

impl Clone for Shape

fn clone(&self) -> Shape

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Shape

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

Formats the value using the given formatter.

impl<'_> From<&'_ [usize]> for Shape

fn from(slice: &[usize]) -> Self

Performs the conversion.

impl From<Vec<usize>> for Shape

fn from(vec: Vec<usize>) -> Self

Performs the conversion.

impl Index<usize> for Shape

type Output = usize

The returned type after indexing.

fn index(&self, i: usize) -> &usize

Performs the indexing (container[index]) operation.

impl IndexMut<usize> for Shape

fn index_mut(&mut self, i: usize) -> &mut usize

Performs the mutable indexing (container[index]) operation.

impl Into<Vec<usize>> for Shape

fn into(self) -> Vec<usize>

Performs the conversion.

impl PartialEq<Shape> for Shape

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

This method tests for self and other values to be equal, and is used by ==.

#[must_use]fn ne(&self, other: &Rhs) -> bool

This method tests for !=.