Struct alumina::shape::NodeShape

pub struct NodeShape { /* fields omitted */ }

Methods

impl NodeShape

pub fn ndims(&self) -> usize

Important traits for &'a [u8]

impl<'a> Read for &'a [u8]impl<'a> Write for &'a mut [u8]

pub fn dimensions(&self) -> &[NodeDim]

Important traits for &'a [u8]

impl<'a> Read for &'a [u8]impl<'a> Write for &'a mut [u8]

pub fn dimensions_mut(&mut self) -> &mut [NodeDim]

pub fn collapse_dimensions_to_minimum(&mut self)

Should be called and only called by operations prior to propagating shape constraints The NodeDim::Interval0 are collapsed to the lower bound, and any NodeDim::Unknown entries will be replaced with 0

pub fn collapse_to_broadcastable_dimension(&self) -> NodeShape

Returns a copy of the shape with all dimensions that arent Known(_) set to 1.

pub fn flat_size(&self) -> NodeDim

pub fn force_flat_size(&self) -> Result<usize>

pub fn to_data_shape(&self) -> Result<IxDyn>

If all dimension values are Known this will This should generally only be called after collapse_dimensions_to_minimum

pub fn is_known(&self) -> bool

pub fn ndim(&self) -> usize

pub fn merge(&self, other: &NodeShape) -> Result<NodeShape>

Trait Implementations

impl Clone for NodeShape

fn clone(&self) -> NodeShape

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for NodeShape

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

Formats the value using the given formatter.

impl PartialEq for NodeShape

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

This method tests for self and other values to be equal, and is used by ==.

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

This method tests for !=.

impl<T: Into<NodeDim> + Clone, I: IntoIterator<Item = T>> From<I> for NodeShape

fn from(i: I) -> NodeShape

Performs the conversion.