Struct Shape 

pub struct Shape { /* private fields */ }

Shape metadata for an N-dimensional tensor.

Stores dimension sizes and precomputed row-major strides. Strides follow the standard row-major formula: strides[n-1] = 1 and strides[i] = strides[i+1] * dims[i+1] for preceding dimensions.

Examples

use grafos_tensor::Shape;

let s = Shape::new(&[2, 3, 4]);
assert_eq!(s.dims(), &[2, 3, 4]);
assert_eq!(s.strides(), &[12, 4, 1]);
assert_eq!(s.numel(), 24);
assert_eq!(s.ndim(), 3);

Implementations

impl Shape

pub fn new(dims: &[usize]) -> Self

Create a new shape from dimension sizes.

Strides are computed in row-major (C) order: the last dimension has stride 1, and each preceding dimension’s stride is the product of all following dimension sizes.

An empty dims slice creates a scalar (0-D) shape with numel() == 1.

Examples

use grafos_tensor::Shape;

let s = Shape::new(&[2, 3]);
assert_eq!(s.strides(), &[3, 1]);

let scalar = Shape::new(&[]);
assert_eq!(scalar.numel(), 1);

pub fn numel(&self) -> usize

Total number of elements in the tensor.

pub fn ndim(&self) -> usize

Number of dimensions.

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

Dimension sizes.

pub fn strides(&self) -> &[usize]

Row-major strides.

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

Formats the value using the given formatter.

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 Eq for Shape

impl StructuralPartialEq for Shape