Struct Tensor 

pub struct Tensor { /* private fields */ }

A multi-dimensional tensor with NHWC layout

Implementations

impl Tensor

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

Create a new tensor with the given shape, initialized to zeros

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

Create a new tensor with the given shape, initialized to ones

pub fn from_data(data: Vec<f32>, shape: &[usize]) -> CnnResult<Self>

Create a tensor from raw data with the given shape

pub fn full(shape: &[usize], value: f32) -> Self

Create a tensor filled with a constant value

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

Get the shape of the tensor

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

Get the strides of the tensor

pub fn ndim(&self) -> usize

Get the number of dimensions

pub fn numel(&self) -> usize

Get the total number of elements

pub fn data(&self) -> &[f32]

Get a reference to the raw data

pub fn data_mut(&mut self) -> &mut [f32]

Get a mutable reference to the raw data

pub fn get_4d(&self, n: usize, h: usize, w: usize, c: usize) -> f32

Get element at index (for 4D NHWC tensor)

pub fn set_4d(&mut self, n: usize, h: usize, w: usize, c: usize, value: f32)

Set element at index (for 4D NHWC tensor)

pub fn batch_size(&self) -> usize

Get batch size (first dimension)

pub fn height(&self) -> usize

Get height (second dimension for NHWC)

pub fn width(&self) -> usize

Get width (third dimension for NHWC)

pub fn channels(&self) -> usize

Get channels (fourth dimension for NHWC)

pub fn reshape(&self, new_shape: &[usize]) -> CnnResult<Self>

Reshape the tensor to a new shape

pub fn view(&self, new_shape: &[usize]) -> CnnResult<Self>

Clone with a new shape (must have same numel)

pub fn slice_batch(&self, start: usize, end: usize) -> CnnResult<Self>

Get a slice of the tensor along the batch dimension

pub fn map<F>(&self, f: F) -> Self

where F: Fn(f32) -> f32,

Apply a function element-wise

pub fn map_inplace<F>(&mut self, f: F)

where F: Fn(f32) -> f32,

Apply a function element-wise in place

pub fn add(&self, other: &Self) -> CnnResult<Self>

Element-wise addition

pub fn mul(&self, other: &Self) -> CnnResult<Self>

Element-wise multiplication

pub fn scale(&self, scalar: f32) -> Self

Scalar multiplication

pub fn sum(&self) -> f32

Sum all elements

pub fn mean(&self) -> f32

Mean of all elements

pub fn max(&self) -> f32

Maximum element

pub fn min(&self) -> f32

Minimum element

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() -> Self

Returns the “default value” for a type.