Struct Tensor 

pub struct Tensor<T: TensorScalar, const N: usize>(/* private fields */);

Rank-N tensor backed by tenferro_tensor::TypedTensor<T>.

Implementations

impl<T: TensorScalar, const N: usize> Tensor<T, N>

pub fn len(&self) -> usize

Total number of elements.

pub fn is_empty(&self) -> bool

Whether the tensor is empty (zero elements).

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

Dimension along axis.

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

All dimensions.

pub fn to_row_major_vec(&self) -> Vec<T>

Export the tensor as a row-major flat vector.

pub fn iter(&self) -> TensorIter<'_, T, N>

Iterate over all elements in row-major order.

pub fn iter_mut(&mut self) -> TensorIterMut<'_, T, N>

Iterate mutably over all elements in row-major order.

pub fn as_inner(&self) -> &TfTensor<T>

Borrow the wrapped tenferro tensor.

pub fn as_inner_mut(&mut self) -> &mut TfTensor<T>

Mutably borrow the wrapped tenferro tensor.

pub fn into_inner(self) -> TfTensor<T>

Consume this wrapper and return the inner tenferro tensor.

pub fn from_tenferro(tensor: TfTensor<T>) -> Self

Wrap an existing tenferro tensor, panicking on rank mismatch.

pub fn from_fn(dims: [usize; N], f: impl FnMut([usize; N]) -> T) -> Self

Create a tensor by applying f to each multi-index (row-major order).

impl<T: TensorScalar, const N: usize> Tensor<T, N>

pub fn from_elem(dims: [usize; N], value: T) -> Self

Create a tensor filled with value.

Trait Implementations

impl<T: TensorScalar, const N: usize> Clone for Tensor<T, N>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: Debug + TensorScalar, const N: usize> Debug for Tensor<T, N>

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

Formats the value using the given formatter.

impl<T: TensorScalar, const N: usize> Index<[usize; N]> for Tensor<T, N>

type Output = T

The returned type after indexing.

fn index(&self, idx: [usize; N]) -> &T

Performs the indexing (container[index]) operation.

impl<T: TensorScalar, const N: usize> IndexMut<[usize; N]> for Tensor<T, N>

fn index_mut(&mut self, idx: [usize; N]) -> &mut T

Performs the mutable indexing (container[index]) operation.

impl<T: TensorScalar + PartialEq, const N: usize> PartialEq for Tensor<T, N>

fn eq(&self, other: &Self) -> 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<T: TensorScalar + Eq, const N: usize> Eq for Tensor<T, N>