Struct Tensor 

pub struct Tensor {
    pub dims: Vec<IoDim>,
}

N-dimensional tensor representation.

A tensor describes the shape and memory layout of FFT data. It consists of a vector of dimensions, each with its own size and strides.

Fields

dims: Vec<IoDim>

The dimensions of the tensor.

Implementations

impl Tensor

pub fn new(dims: Vec<IoDim>) -> Self

Create a new tensor with the given dimensions.

pub fn empty() -> Self

Create an empty (rank-0) tensor.

pub fn rank1(n: usize) -> Self

Create a 1D contiguous tensor.

pub fn rank2(n0: usize, n1: usize) -> Self

Create a 2D row-major contiguous tensor.

pub fn rank3(n0: usize, n1: usize, n2: usize) -> Self

Create a 3D row-major contiguous tensor.

pub fn rank(&self) -> usize

Get the rank (number of dimensions).

pub fn is_empty(&self) -> bool

Check if the tensor is empty (rank 0).

pub fn total_size(&self) -> usize

Get the total number of elements.

pub fn is_contiguous(&self) -> bool

Check if all dimensions are contiguous.

pub fn is_inplace_compatible(&self) -> bool

Check if input and output have identical strides (in-place compatible).

pub fn split(&self, axis: usize) -> (Self, Self)

Split the tensor at the given axis.

Returns (outer dimensions, inner dimensions).

pub fn first(&self) -> Option<&IoDim>

Get the first dimension, if any.

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

Get the last dimension, if any.

pub fn pop_front(&self) -> Option<(IoDim, Self)>

Remove and return the first dimension.

pub fn push(&mut self, dim: IoDim)

Append a dimension.

pub fn push_front(&mut self, dim: IoDim)

Prepend a dimension.

Trait Implementations

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Tensor

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

Formats the value using the given formatter.

impl Default for Tensor

fn default() -> Tensor

Returns the “default value” for a type.

impl Hash for Tensor

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

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 Tensor

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

impl StructuralPartialEq for Tensor