Struct TensorShape 

pub struct TensorShape {
    pub dims: Vec<Option<usize>>,
}

Tensor shape information for cost estimation.

Fields

dims: Vec<Option<usize>>

Dimension sizes (None = unknown)

Implementations

impl TensorShape

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

Create a new tensor shape.

pub fn num_elements(&self) -> Option<usize>

Get the number of elements (if all dimensions are known).

pub fn rank(&self) -> usize

Get the rank (number of dimensions).

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.