Struct TensorShape 

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

A multi-dimensional tensor shape supporting mixed static/dynamic dimensions.

For example, [batch, 224, 224, 3] where batch is dynamic and spatial dimensions are fixed.

Fields

dims: Vec<Dimension>

The dimensions of the tensor.

Implementations

impl TensorShape

pub fn fixed(sizes: &[u32]) -> Self

Create a shape where all dimensions are fixed.

pub fn all_dynamic(rank: usize) -> Self

Create a shape where all dimensions are dynamic (unnamed).

pub fn rank(&self) -> usize

Returns the number of dimensions (rank).

pub fn is_fully_static(&self) -> bool

Returns true if all dimensions are statically known.

pub fn is_fully_dynamic(&self) -> bool

Returns true if all dimensions are dynamic (unnamed).

pub fn has_dynamic_dims(&self) -> bool

Returns true if any dimension is not statically known (either dynamic or symbolic).

pub fn is_mixed(&self) -> bool

Returns true if the shape contains a mix of static and dynamic/symbolic dims.

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 Eq for TensorShape

impl Hash for TensorShape

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 TensorShape

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for TensorShape