Struct Array3 

pub struct Array3<T> { /* private fields */ }

Owned 3D row-major array.

Implementations

impl<T> Array3<T>

pub fn from_vec(shape: [usize; 3], data: Vec<T>) -> Result<Self>

Build an array from row-major data.

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

Build an array from a function over (i, j, k).

pub fn try_from_fn( shape: [usize; 3], f: impl FnMut(usize, usize, usize) -> T, ) -> Result<Self>

Fallibly build an array from a function over (i, j, k).

pub fn shape(&self) -> [usize; 3]

Shape as [dim0, dim1, dim2].

pub fn strides(&self) -> [isize; 3]

Row-major strides in elements.

pub fn len(&self) -> usize

Number of elements.

pub fn is_empty(&self) -> bool

Whether the array has zero elements.

pub fn as_slice(&self) -> &[T]

Borrow the backing slice.

pub fn as_mut_slice(&mut self) -> &mut [T]

Borrow the backing slice mutably.

pub fn view(&self) -> ArrayView3<'_, T>

Immutable strided view.

pub fn view_mut(&mut self) -> ArrayViewMut3<'_, T>

Mutable strided view.

pub fn get(&self, i: usize, j: usize, k: usize) -> Option<&T>

Get an element reference.

pub fn get_mut(&mut self, i: usize, j: usize, k: usize) -> Option<&mut T>

Get a mutable element reference.

pub fn matrix_at(&self, axis: Axis3, index: usize) -> Result<ArrayView2<'_, T>>

Extract a 2D matrix view by fixing one axis.

pub fn for_each_matrix( &self, axis: Axis3, f: impl FnMut(usize, ArrayView2<'_, T>) -> Result<()>, ) -> Result<()>

Visit each 2D matrix slice along axis in order.

pub fn matrix_at_mut( &mut self, axis: Axis3, index: usize, ) -> Result<ArrayViewMut2<'_, T>>

Extract a mutable 2D matrix view by fixing one axis.

pub fn for_each_matrix_mut( &mut self, axis: Axis3, f: impl FnMut(usize, ArrayViewMut2<'_, T>) -> Result<()>, ) -> Result<()>

Visit each mutable 2D matrix slice along axis in order.

impl<T: Clone> Array3<T>

pub fn filled(shape: [usize; 3], value: T) -> Self

Fill a new array with value.

pub fn try_filled(shape: [usize; 3], value: T) -> Result<Self>

Fallibly fill a new array with value.

pub fn clone_contiguous(view: ArrayView3<'_, T>) -> Self

Clone into compact row-major storage.

pub fn unfold(&self, axis: Axis3) -> Array2<T>

Unfold the tensor along axis into a row-major matrix.

For shape [i, j, k], mode-0 unfolding has shape [i, j * k], mode-1 has shape [j, i * k], and mode-2 has shape [k, i * j].

pub fn fold( axis: Axis3, shape: [usize; 3], matrix: ArrayView2<'_, T>, ) -> Result<Self>

Fold a mode-unfolded matrix back into a tensor with shape.

impl<T: Float> Array3<T>

pub fn zeros(shape: [usize; 3]) -> Self

Array filled with zeros.

pub fn try_zeros(shape: [usize; 3]) -> Result<Self>

Fallibly allocate an array filled with zeros.

pub fn ones(shape: [usize; 3]) -> Self

Array filled with ones.

pub fn try_ones(shape: [usize; 3]) -> Result<Self>

Fallibly allocate an array filled with ones.

pub fn zeros_like(&self) -> Self

Allocate another array with the same shape, filled with zeros.

pub fn scale(&mut self, alpha: T)

Scale in place.

pub fn scaled(&self, alpha: T) -> Self

Return self * alpha without modifying self.

pub fn scaled_into(&self, alpha: T, out: ArrayViewMut3<'_, T>) -> Result<()>

Write self * alpha into out without modifying self.

pub fn add(&self, other: ArrayView3<'_, T>) -> Result<Self>

Return self + other without modifying either input.

pub fn add_into( &self, other: ArrayView3<'_, T>, out: ArrayViewMut3<'_, T>, ) -> Result<()>

Write self + other into out without modifying either input.

pub fn sub(&self, other: ArrayView3<'_, T>) -> Result<Self>

Return self - other without modifying either input.

pub fn sub_into( &self, other: ArrayView3<'_, T>, out: ArrayViewMut3<'_, T>, ) -> Result<()>

Write self - other into out without modifying either input.

pub fn mul(&self, other: ArrayView3<'_, T>) -> Result<Self>

Return the elementwise product self * other without modifying either input.

pub fn mul_into( &self, other: ArrayView3<'_, T>, out: ArrayViewMut3<'_, T>, ) -> Result<()>

Write the elementwise product into out without modifying either input.

pub fn hadamard(&self, other: ArrayView3<'_, T>) -> Result<Self>

Return the Hadamard product without modifying either input.

pub fn hadamard_into( &self, other: ArrayView3<'_, T>, out: ArrayViewMut3<'_, T>, ) -> Result<()>

Write the Hadamard product into out without modifying either input.

pub fn div(&self, other: ArrayView3<'_, T>) -> Result<Self>

Return the elementwise quotient self / other without modifying either input.

pub fn div_into( &self, other: ArrayView3<'_, T>, out: ArrayViewMut3<'_, T>, ) -> Result<()>

Write the elementwise quotient into out without modifying either input.

pub fn axpy_result(&self, alpha: T, x: ArrayView3<'_, T>) -> Result<Self>

Return self + alpha * x without modifying either input.

pub fn axpy_into( &self, alpha: T, x: ArrayView3<'_, T>, out: ArrayViewMut3<'_, T>, ) -> Result<()>

Write self + alpha * x into out without modifying either input.

pub fn add_assign_view(&mut self, other: ArrayView3<'_, T>) -> Result<()>

In-place addition.

pub fn sub_assign_view(&mut self, other: ArrayView3<'_, T>) -> Result<()>

In-place subtraction.

pub fn mul_assign_view(&mut self, other: ArrayView3<'_, T>) -> Result<()>

Hadamard product in place.

pub fn div_assign_view(&mut self, other: ArrayView3<'_, T>) -> Result<()>

In-place elementwise division.

pub fn axpy(&mut self, alpha: T, x: ArrayView3<'_, T>) -> Result<()>

Compute self += alpha * x.

pub fn zip_map( &self, other: ArrayView3<'_, T>, f: impl FnMut(T, T) -> T, ) -> Result<Self>

Return an elementwise zip-map without modifying either input.

pub fn zip_map_into( &self, other: ArrayView3<'_, T>, out: ArrayViewMut3<'_, T>, f: impl FnMut(T, T) -> T, ) -> Result<()>

Write an elementwise zip-map into out without modifying either input.

pub fn map_inplace(&mut self, f: impl FnMut(T) -> T)

Map values in place.

pub fn zip_map_inplace( &mut self, other: ArrayView3<'_, T>, f: impl FnMut(T, T) -> T, ) -> Result<()>

Zip-map another view into this array in place.

pub fn fill_uniform(&mut self, seed: u64)

Fill with deterministic uniform random values in [0, 1).

pub fn fill_randn(&mut self, seed: u64)

Fill with deterministic standard-normal random values.

pub fn sum(&self) -> T

Sum all elements.

pub fn mean(&self) -> T

Mean of all elements, or zero for an empty array.

pub fn norm_frobenius(&self) -> T

Frobenius norm.

pub fn max_abs(&self) -> T

Maximum absolute element, or zero for an empty array.

pub fn dot(&self, other: ArrayView3<'_, T>) -> Result<T>

Dot product of two tensors treated as flattened row-major buffers.

Trait Implementations

impl<T: Clone> Clone for Array3<T>

fn clone(&self) -> Array3<T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: Debug> Debug for Array3<T>

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

Formats the value using the given formatter.

impl<T> Index<(usize, usize, usize)> for Array3<T>

type Output = T

The returned type after indexing.

fn index(&self, index: (usize, usize, usize)) -> &Self::Output

Performs the indexing (container[index]) operation.

impl<T> IndexMut<(usize, usize, usize)> for Array3<T>

fn index_mut(&mut self, index: (usize, usize, usize)) -> &mut Self::Output

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

impl<T: PartialEq> PartialEq for Array3<T>

fn eq(&self, other: &Array3<T>) -> 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> StructuralPartialEq for Array3<T>