Struct ShapeEnv 

pub struct ShapeEnv { /* private fields */ }

Maps symbolic dimension names to concrete values.

Used during shape resolution to convert symbolic shapes to concrete ones.

Examples

let mut env = ShapeEnv::new();
env.bind("Batch", 32);
env.bind("SeqLen", 128);
assert_eq!(env.get("Batch"), Some(32));

Implementations

impl ShapeEnv

pub fn new() -> Self

Create an empty shape environment.

pub fn bind(&mut self, name: impl Into<String>, value: usize)

Bind a symbolic name to a concrete value.

pub fn get(&self, name: &str) -> Option<usize>

Look up a symbolic name.

pub fn is_bound(&self, name: &str) -> bool

Check if a name is bound.

pub fn bindings(&self) -> &HashMap<String, usize>

Get all bindings.

pub fn len(&self) -> usize

Number of bindings.

pub fn is_empty(&self) -> bool

Is the environment empty?

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

Merge another environment into this one. Returns an error if there are conflicting bindings.

Trait Implementations

impl Clone for ShapeEnv

fn clone(&self) -> ShapeEnv

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ShapeEnv

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

Formats the value using the given formatter.

impl Default for ShapeEnv

fn default() -> Self

Returns the “default value” for a type.

impl From<&[(&str, usize)]> for ShapeEnv

fn from(bindings: &[(&str, usize)]) -> Self

Converts to this type from the input type.