Struct Shape 

pub struct Shape(pub Vec<usize>);

Shape of a tensor (dimensions).

Tuple Fields

0: Vec<usize>

Implementations

impl Shape

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

Creates a new shape.

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

Returns the dimensions.

pub fn ndim(&self) -> usize

Returns the number of dimensions.

pub fn numel(&self) -> usize

Returns the total number of elements.

pub fn broadcast_compatible(&self, other: &Self) -> bool

Checks if shapes are broadcast compatible.

pub fn broadcast_shape(&self, other: &Self) -> Option<Self>

Computes broadcast shape.

Trait Implementations

impl Clone for Shape

fn clone(&self) -> Shape

Returns a duplicate 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(dims: &[usize]) -> Self

Converts to this type from the input type.

impl From<Vec<usize>> for Shape

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

Converts to this type from the input type.

impl Hash for Shape

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for Shape

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

Tests for self and other values to be equal, and is used by ==.

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

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

impl Eq for Shape

impl StructuralPartialEq for Shape