Struct microtensor::Shape

pub struct Shape {
    pub dims: Vec<usize>,
    /* private fields */
}

The shape of a Tensor.

Fields

dims: Vec<usize>

Implementations

impl Shape

pub fn new(dims: &[usize]) -> Self

pub fn strided(dims: &[usize], strides: &[isize]) -> Self

pub fn offset(dims: &[usize], strides: &[isize], offset: usize) -> Self

pub fn size(&self) -> usize

pub fn rank(&self) -> usize

pub fn contiguous(&self) -> bool

pub fn at_or(&self, idx: isize, or: usize) -> usize

pub fn iter(&self) -> Box<dyn Iterator<Item = usize> + '_>

pub fn view(&self, shape: &[usize]) -> Self

pub fn range(&self, ranges: &[Range<isize>]) -> Self

pub fn step(&self, steps: &[isize]) -> Self

pub fn squeeze(&self, squeezed: &[isize]) -> Self

pub fn squeeze_first(&self, n: usize) -> Self

pub fn squeeze_all(&self) -> Self

pub fn unsqueeze(&self, dim: isize) -> Self

pub fn extend_front(&self, size: usize) -> Self

pub fn broadcast(&self, other: &Self, ignore_from: Option<isize>) -> Self

pub fn transpose(&self, dim1: isize, dim2: isize) -> Self

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<'de> Deserialize<'de> for Shape

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for Shape

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

Formats the value using the given formatter.

impl Index<isize> for Shape

type Output = usize

The returned type after indexing.

fn index<'a>(&'a self, i: isize) -> &'a usize

Performs the indexing (container[index]) operation.

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 ==.

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

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for Shape

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for Shape