Struct Tensor 

pub struct Tensor<T: Scalar> { /* private fields */ }

An N-dimensional array of numeric values.

Tensors are the core data structure for all computations in Axonml. They support arbitrary dimensions, automatic broadcasting, and efficient memory sharing between views.

Implementations

impl<T: Scalar> Tensor<T>

pub fn from_storage(storage: Storage<T>, shape: &[usize]) -> Result<Self>

Creates a new tensor from storage with the given shape.

Arguments

• storage - The underlying data storage
• shape - Shape of the tensor

Returns

New tensor, or error if shape doesn’t match storage size.

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

Creates a new tensor from a vector with the given shape.

Arguments

• data - Vector of data
• shape - Shape of the tensor

Returns

New tensor, or error if shape doesn’t match data length.

pub fn from_slice(data: &[T], shape: &[usize]) -> Result<Self>

Creates a new tensor from a slice with the given shape.

Arguments

• data - Slice of data to copy
• shape - Shape of the tensor

Returns

New tensor, or error if shape doesn’t match data length.

pub fn scalar(value: T) -> Self

Creates a scalar tensor (0-dimensional).

Arguments

• value - The scalar value

Returns

New 0-dimensional tensor.

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

Creates a tensor filled with zeros.

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

where T: Numeric,

Creates a tensor filled with ones.

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

Creates a tensor filled with a constant value.

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

where T: Float, StandardNormal: Distribution<T>,

Creates a tensor with random values from standard normal distribution.

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

where T: Float, Standard: Distribution<T>,

Creates a tensor with random values from uniform distribution [0, 1).

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

Returns the shape of the tensor.

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

Returns the strides of the tensor.

pub fn ndim(&self) -> usize

Returns the number of dimensions.

pub fn numel(&self) -> usize

Returns the total number of elements.

pub fn is_empty(&self) -> bool

Returns true if the tensor is empty (has zero elements).

pub fn size(&self, dim: i64) -> Result<usize>

Returns the size of a specific dimension.

Arguments

• dim - Dimension index (supports negative indexing)

pub fn device(&self) -> Device

Returns the device this tensor is on.

pub fn is_contiguous(&self) -> bool

Returns true if the tensor is contiguous in memory.

pub fn is_scalar(&self) -> bool

Returns true if this tensor is a scalar (0-dimensional).

pub fn get(&self, indices: &[usize]) -> Result<T>

Returns the element at the given indices.

Arguments

• indices - Multi-dimensional indices

pub fn set(&self, indices: &[usize], value: T) -> Result<()>

Sets the element at the given indices.

Arguments

• indices - Multi-dimensional indices
• value - Value to set

pub fn item(&self) -> Result<T>

Returns the scalar value for a 0-dimensional tensor.

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

Returns the data as a contiguous vector.

If the tensor is already contiguous, this returns a reference. Otherwise, it copies the data into a new contiguous vector.

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

Returns a new tensor with the specified shape.

The total number of elements must remain the same. Supports -1 in one dimension to infer the size.

Arguments

• new_shape - Target shape

pub fn flatten(&self) -> Self

Returns a new tensor with a flattened shape.

pub fn squeeze(&self, dim: Option<i64>) -> Result<Self>

Returns a new tensor with dimensions of size 1 removed.

Arguments

• dim - Optional specific dimension to squeeze

pub fn unsqueeze(&self, dim: i64) -> Result<Self>

Returns a new tensor with a dimension of size 1 inserted.

Arguments

• dim - Position to insert the new dimension

pub fn transpose(&self, dim0: i64, dim1: i64) -> Result<Self>

Transposes two dimensions.

Arguments

• dim0 - First dimension
• dim1 - Second dimension

pub fn t(&self) -> Result<Self>

Returns the transpose of a 2D tensor.

pub fn permute(&self, dims: &[usize]) -> Result<Self>

Returns a permuted tensor with dimensions reordered.

Arguments

• dims - New order of dimensions

pub fn contiguous(&self) -> Self

Returns a contiguous copy of the tensor.

pub fn to_device(&self, device: Device) -> Result<Self>

Transfers the tensor to a different device.

Arguments

• device - Target device

pub fn cpu(&self) -> Result<Self>

Transfers to CPU.

pub fn clone_deep(&self) -> Self

Creates a deep copy of this tensor with its own storage.

impl<T: Numeric> Tensor<T>

pub fn fill_(&self, value: T)

Fills the tensor with a value.

pub fn zero_(&self)

Fills the tensor with zeros.

pub fn sum(&self) -> Self

Returns the sum of all elements.

pub fn prod(&self) -> Self

Returns the product of all elements.

pub fn max(&self) -> Result<Self>

Returns the maximum element.

pub fn min(&self) -> Result<Self>

Returns the minimum element.

pub fn argmax(&self) -> Result<usize>

Returns the index of the maximum element.

pub fn argmin(&self) -> Result<usize>

Returns the index of the minimum element.

impl<T: Float> Tensor<T>

pub fn mean(&self) -> Result<Self>

Returns the mean of all elements.

pub fn relu(&self) -> Self

Applies ReLU activation: max(0, x).

pub fn sigmoid(&self) -> Self

Applies sigmoid activation: 1 / (1 + exp(-x)).

pub fn tanh(&self) -> Self

Applies tanh activation.

pub fn exp(&self) -> Self

Applies exponential function.

pub fn ln(&self) -> Self

Applies natural logarithm.

pub fn sqrt(&self) -> Self

Applies square root.

pub fn pow(&self, exp: T) -> Self

Computes element-wise power.

pub fn gelu(&self) -> Self

GELU activation function (Gaussian Error Linear Unit).

pub fn silu(&self) -> Self

SiLU/Swish activation function.

pub fn softmax(&self, dim: i32) -> Self

Softmax along specified dimension.

pub fn log_softmax(&self, dim: i32) -> Self

Log softmax along specified dimension.

pub fn mean_dim(&self, dim: i32, keepdim: bool) -> Self

Mean along a dimension.

pub fn var_dim(&self, dim: i32, keepdim: bool) -> Self

Variance along a dimension.

pub fn broadcast_to(&self, shape: &[usize]) -> Self

Broadcasts tensor to a new shape.

pub fn slice(&self, ranges: &[Range<usize>]) -> Self

Slices the tensor using ranges for each dimension.

impl<T: Numeric> Tensor<T>

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

Element-wise addition with broadcasting.

pub fn sub(&self, other: &Self) -> Result<Self>

Element-wise subtraction with broadcasting.

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

Element-wise multiplication with broadcasting.

pub fn div(&self, other: &Self) -> Result<Self>

Element-wise division with broadcasting.

pub fn add_scalar(&self, scalar: T) -> Self

Scalar addition.

pub fn mul_scalar(&self, scalar: T) -> Self

Scalar multiplication.

pub fn neg(&self) -> Self

Element-wise negation.

pub fn matmul(&self, other: &Self) -> Result<Self>

Matrix multiplication with batching support.

Supports:

• 2D @ 2D: [m, k] @ [k, n] -> [m, n]
• 3D @ 3D: [batch, m, k] @ [batch, k, n] -> [batch, m, n]
• 4D @ 4D: [b1, b2, m, k] @ [b1, b2, k, n] -> [b1, b2, m, n]

pub fn dot(&self, other: &Self) -> Result<Self>

Dot product for 1D tensors.

impl<T: Scalar> Tensor<T>

pub fn slice_dim0(&self, start: usize, end: usize) -> Result<Self>

Returns a slice of the tensor along the first dimension.

Arguments

• start - Start index (inclusive)
• end - End index (exclusive)

pub fn select(&self, dim: usize, index: usize) -> Result<Self>

Returns a view selecting a single index along a dimension.

This reduces the dimensionality by 1.

Arguments

• dim - Dimension to select from
• index - Index to select

pub fn narrow(&self, dim: usize, start: usize, length: usize) -> Result<Self>

Returns a narrow view along a dimension.

Arguments

• dim - Dimension to narrow
• start - Start index
• length - Length of the narrow view

pub fn chunk(&self, chunks: usize, dim: usize) -> Result<Vec<Self>>

Splits the tensor into chunks along a dimension.

Arguments

• chunks - Number of chunks
• dim - Dimension to split along

pub fn split(&self, sizes: &[usize], dim: usize) -> Result<Vec<Self>>

Splits the tensor into parts of specified sizes along a dimension.

Arguments

• sizes - Size of each part
• dim - Dimension to split along

impl<T: Numeric> Tensor<T>

pub fn gather(&self, dim: usize, indices: &Tensor<i64>) -> Result<Self>

Gathers values along a dimension according to indices.

Arguments

• dim - Dimension to gather along
• indices - Indices tensor

pub fn masked_select(&self, mask: &[bool]) -> Result<Self>

Returns elements selected by a boolean mask.

Arguments

• mask - Boolean mask tensor

pub fn masked_fill_(&self, mask: &[bool], value: T) -> Result<()>

Sets elements according to a boolean mask.

Arguments

• mask - Boolean mask tensor
• value - Value to set where mask is true

Trait Implementations

impl<T: Numeric> Add<T> for &Tensor<T>

type Output = Tensor<T>

The resulting type after applying the + operator.

fn add(self, scalar: T) -> Self::Output

Performs the + operation.

impl<T: Numeric> Add for &Tensor<T>

type Output = Tensor<T>

The resulting type after applying the + operator.

fn add(self, other: Self) -> Self::Output

Performs the + operation.

impl<T: Clone + Scalar> Clone for Tensor<T>

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

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<T: Scalar + Display> Debug for Tensor<T>

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

Formats the value using the given formatter.

impl<T: Scalar + Display> Display for Tensor<T>

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

Formats the value using the given formatter.

impl<T: Numeric> Div for &Tensor<T>

type Output = Tensor<T>

The resulting type after applying the / operator.

fn div(self, other: Self) -> Self::Output

Performs the / operation.

impl<T: Numeric> Mul<T> for &Tensor<T>

type Output = Tensor<T>

The resulting type after applying the * operator.

fn mul(self, scalar: T) -> Self::Output

Performs the * operation.

impl<T: Numeric> Mul for &Tensor<T>

type Output = Tensor<T>

The resulting type after applying the * operator.

fn mul(self, other: Self) -> Self::Output

Performs the * operation.

impl<T: Numeric> Neg for &Tensor<T>

type Output = Tensor<T>

The resulting type after applying the - operator.

fn neg(self) -> Self::Output

Performs the unary - operation.

impl<T: Numeric> Sub for &Tensor<T>

type Output = Tensor<T>

The resulting type after applying the - operator.

fn sub(self, other: Self) -> Self::Output

Performs the - operation.