Struct Shape 

pub struct Shape { /* private fields */ }

Tensor shape: ordered list of dimensions + element type.

SmallVec<[Dim; 4]> avoids heap allocation for up to 4D tensors (the common case).

Implementations

impl Shape

pub fn batch_seq(batch: u32, seq: u32, hidden: usize, dtype: DType) -> Shape

[batch, seq, hidden] with symbolic leading axes.

pub fn batch_seq_2d(batch: u32, seq: u32, dtype: DType) -> Shape

[batch, seq] matrix.

pub fn batch_seq_heads( batch: u32, seq: u32, heads: usize, head_dim: usize, dtype: DType, ) -> Shape

[batch, seq, heads, head_dim] attention layout.

impl Shape

pub fn new(dims: &[usize], dtype: DType) -> Shape

Create a shape from static dimensions.

pub fn from_dims(dims: &[Dim], dtype: DType) -> Shape

Create a shape with mixed static/dynamic dimensions.

pub fn scalar(dtype: DType) -> Shape

Scalar (0-dimensional).

pub fn rank(&self) -> usize

pub fn dtype(&self) -> DType

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

pub fn dim(&self, i: usize) -> Dim

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

Set of dynamic dim symbols this shape references. Useful for “what bindings does this graph need?” queries on inputs.

pub fn bind(&self, bindings: &DimBinding) -> Shape

Specialize the shape against a binding (symbol → static size). Unknown symbols stay Dim::Dynamic. Plan #54: the step that takes a “compile once, run at any seq length” graph and produces the runtime-specific concrete shape.

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

Total number of elements (only if all dims are static).

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

Total size in bytes (only if all dims are static).

pub fn is_static(&self) -> bool

True if all dimensions are statically known.

pub fn with_dim(self, axis: usize, dim: Dim) -> Shape

Replace a dimension.

pub fn with_dtype(self, dtype: DType) -> Shape

Change dtype (for cast operations).

pub fn broadcast_with(&self, other: &Shape) -> Result<Shape, String>

Numpy-style broadcast with another shape (fusion / lowering).

Trait Implementations

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<'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 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 PartialEq for Shape

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

impl StructuralPartialEq for Shape