Struct Shape 

pub struct Shape { /* private fields */ }

Shape represents the dimensions of a tensor using u32 for memory efficiency.

This struct provides a type-safe wrapper around dimension vectors and includes utilities for shape manipulation and validation.

Implementations

impl Shape

pub fn new(dims: &[u32]) -> Shape

Creates a new shape from a slice of dimensions.

pub fn scalar() -> Shape

Creates a scalar shape (0 dimensions).

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

Returns the dimensions as a slice.

pub fn dims_mut(&mut self) -> &mut Vec<u32>

Returns a mutable reference to the dimensions.

pub fn to_vec(&self) -> Vec<u32>

Converts the shape to a Vec.

pub fn last(&self) -> Option<u32>

Returns the last dimension, or None if the shape is scalar.

pub fn ndim(&self) -> u32

Returns the number of dimensions (rank).

pub fn dim(&self, index: u32) -> Option<u32>

Returns the size of a specific dimension.

pub fn size(&self) -> u32

Returns the total number of elements.

pub fn is_scalar(&self) -> bool

Returns true if this is a scalar (0 dimensions).

pub fn can_broadcast_to(&self, target: &Shape) -> bool

Checks if this shape can be broadcast to the target shape.

pub fn broadcast_shape(lhs: &Shape, rhs: &Shape) -> Option<Shape>

Computes the broadcast shape between two shapes.

pub fn normalize_axis(&self, axis: i32) -> Option<u32>

Normalizes a dimension index, handling negative indices.

Trait Implementations

impl AsRef<[u32]> for Shape

fn as_ref(&self) -> &[u32]

Converts this type into a shared reference of the (usually inferred) input type.

impl<'__de, __Context> BorrowDecode<'__de, __Context> for Shape

fn borrow_decode<__D>(decoder: &mut __D) -> Result<Shape, DecodeError>

where __D: BorrowDecoder<'__de, Context = __Context>,

Attempt to decode this type with the given BorrowDecode.

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<(), Error>

Formats the value using the given formatter.

impl<__Context> Decode<__Context> for Shape

fn decode<__D>(decoder: &mut __D) -> Result<Shape, DecodeError>

where __D: Decoder<Context = __Context>,

Attempt to decode this type with the given Decode.

impl<'de> Deserialize<'de> for Shape

fn deserialize<__D>( __deserializer: __D, ) -> Result<Shape, <__D as Deserializer<'de>>::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for Shape

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

Formats the value using the given formatter.

impl Encode for Shape

fn encode<__E>(&self, encoder: &mut __E) -> Result<(), EncodeError>

where __E: Encoder,

Encode a given type.

impl Eq for Shape

impl From<&Shape> for Shape

fn from(shape: &Shape) -> Shape

Converts to this type from the input type.

impl From<&Vec<u32>> for Shape

fn from(dims: &Vec<u32>) -> Shape

Converts to this type from the input type.

impl From<&Vec<usize>> for Shape

fn from(dims: &Vec<usize>) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<&[i8; N]> for Shape

fn from(dims: &[i8; N]) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<&[i32; N]> for Shape

fn from(dims: &[i32; N]) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<&[isize; N]> for Shape

fn from(dims: &[isize; N]) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<&[u8; N]> for Shape

fn from(dims: &[u8; N]) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<&[u32; N]> for Shape

fn from(dims: &[u32; N]) -> Shape

Converts to this type from the input type.

impl From<&[u32]> for Shape

fn from(dims: &[u32]) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<&[usize; N]> for Shape

fn from(dims: &[usize; N]) -> Shape

Converts to this type from the input type.

impl From<&[usize]> for Shape

fn from(dims: &[usize]) -> Shape

Converts to this type from the input type.

impl From<Vec<u32>> for Shape

fn from(dims: Vec<u32>) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<[i8; N]> for Shape

fn from(dims: [i8; N]) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<[i32; N]> for Shape

fn from(dims: [i32; N]) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<[isize; N]> for Shape

fn from(dims: [isize; N]) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<[u8; N]> for Shape

fn from(dims: [u8; N]) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<[u32; N]> for Shape

fn from(dims: [u32; N]) -> Shape

Converts to this type from the input type.

impl<const N: usize> From<[usize; N]> for Shape

fn from(dims: [usize; N]) -> Shape

Converts to this type from the input type.

impl Hash for Shape

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

where __H: Hasher,

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 Index<u32> for Shape

type Output = u32

The returned type after indexing.

fn index(&self, index: u32) -> &<Shape as Index<u32>>::Output

Performs the indexing (container[index]) operation.

impl PartialEq for Shape

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

Equality operator ==.

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

Inequality operator !=.

impl Serialize for Shape

fn serialize<__S>( &self, __serializer: __S, ) -> Result<<__S as Serializer>::Ok, <__S as Serializer>::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for Shape