Struct TensorShape 

pub struct TensorShape {
    pub dims: Vec<DimSize>,
}

Tensor shape representation

Fields

dims: Vec<DimSize>

Implementations

impl TensorShape

pub fn new(dims: Vec<DimSize>) -> Self

pub fn static_shape(sizes: Vec<usize>) -> Self

pub fn dynamic(rank: usize) -> Self

pub fn rank(&self) -> usize

pub fn is_static(&self) -> bool

pub fn as_static(&self) -> Option<Vec<usize>>

pub fn compatible_with(&self, other: &TensorShape) -> bool

Check if two shapes are compatible (can broadcast or are equal)

Trait Implementations

impl Clone for TensorShape

fn clone(&self) -> TensorShape

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorShape

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

Formats the value using the given formatter.

impl PartialEq for TensorShape

fn eq(&self, other: &TensorShape) -> 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 TensorShape

impl StructuralPartialEq for TensorShape