Struct DenseTensor 

pub struct DenseTensor { /* private fields */ }

密集张量：N 维数组的高性能实现

使用 64 字节对齐，支持 BLAS 加速的矩阵运算

Implementations

impl DenseTensor

pub fn nbytes(&self) -> usize

获取字节大小

pub fn is_contiguous(&self) -> bool

检查是否连续存储

pub fn alignment(&self) -> usize

获取对齐字节数

pub fn new(data: Vec<f64>, shape: Vec<usize>) -> Self

创建新的密集张量

Arguments

• data - 数据向量（行优先顺序，C-order）
• shape - 张量形状

Returns

返回新创建的 DenseTensor

Panics

如果 data 长度与 shape 不匹配会 panic

pub fn from_vec(data: Vec<f64>, shape: Vec<usize>) -> Self

从 Vec 创建张量（行优先顺序）

pub fn zeros(shape: Vec<usize>) -> Self

创建全零张量

pub fn ones(shape: Vec<usize>) -> Self

创建全一张量

pub fn scalar(value: f64) -> Self

创建标量张量

pub fn matrix(rows: usize, cols: usize, data: Vec<f64>) -> Self

创建 2D 矩阵

pub fn eye(size: usize) -> Self

创建 2D 单位矩阵

pub fn from_ndarray(arr: &Array2<f64>) -> Self

从 ndarray::Array2 创建

pub fn to_ndarray(&self) -> Result<Array2<f64>, TensorError>

转换为 ndarray::Array2

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

获取数据切片

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

获取可变数据切片

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

获取 strides

pub fn is_c_contiguous(&self) -> bool

检查是否为连续内存（C-order）

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

获取指定索引的元素

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

设置指定索引的元素

pub fn row(&self, row: usize) -> Result<Vec<f64>, TensorError>

获取指定行的数据

pub fn col(&self, col: usize) -> Result<Vec<f64>, TensorError>

获取指定列的数据

impl DenseTensor

pub fn silu(&self) -> Self

SiLU activation: f(x) = x * sigmoid(x)

pub fn gelu_derivative(&self) -> Self

GELU derivative (for backpropagation)

pub fn mean_dim(&self, dim: isize) -> Self

Mean along a specific dimension

pub fn var_dim(&self, dim: isize) -> Self

Variance along a specific dimension

pub fn sqrt(&self) -> Self

Element-wise square root

pub fn neg(&self) -> Self

Negate the tensor

pub fn gt(&self, value: f64) -> Self

Element-wise greater than comparison (returns 1.0 if true, 0.0 otherwise)

pub fn mask_fill(&self, mask: &Self, value: f64) -> Self

Fill values with a given value where mask is 1.0

pub fn transpose_2d(&self) -> Self

Transpose for 2D tensors (convenience method)

pub fn get_row(&self, row: usize) -> Self

Get a row from a 2D or 3D tensor

pub fn set_row(&mut self, row: usize, data: &Self)

Set a row in the tensor (mutable)

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

Create a tensor filled with a value

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

Scale the tensor by a scalar

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

Softmax along the last dimension

pub fn relu(&self) -> Self

ReLU activation

pub fn gelu(&self) -> Self

GELU activation

pub fn cos(&self) -> Self

Element-wise cosine

pub fn sin(&self) -> Self

Element-wise sine

pub fn ln(&self) -> Self

Element-wise natural logarithm

pub fn bmm_broadcast_weight(&self, weight: &DenseTensor) -> Self

Batched matrix multiplication For 3D tensors: [batch, seq, hidden] @ [hidden, out] -> [batch, seq, out] Broadcasts 2D weight across batch dimension

pub fn expand_last_dim(&self, target_dim: usize) -> Self

Expand the last dimension from 1 to target_dim (for broadcasting) E.g., [batch, seq, 1] -> [batch, seq, target_dim]

pub fn expand_to_3d(&self, batch: usize, seq: usize) -> Self

Expand a 1D tensor [hidden] to 3D [batch, seq, hidden]

pub fn expand_last_dim_2d(&self, target_dim: usize) -> Self

Expand the last dimension from 1 to target_dim for 2D tensors E.g., [seq, 1] -> [seq, target_dim]

pub fn expand_to_2d(&self, seq: usize) -> Self

Expand a 1D tensor [hidden] to 2D [seq, hidden]

Trait Implementations

impl Aggregator<DenseTensor> for SumAggregator

Available on crate feature tensor-gnn only.

fn aggregate(&self, messages: &[DenseTensor]) -> DenseTensor

聚合消息

impl Aggregator<DenseTensor> for MeanAggregator

Available on crate feature tensor-gnn only.

fn aggregate(&self, messages: &[DenseTensor]) -> DenseTensor

聚合消息

impl Aggregator<DenseTensor> for MaxAggregator

Available on crate feature tensor-gnn only.

fn aggregate(&self, messages: &[DenseTensor]) -> DenseTensor

聚合消息

impl Clone for DenseTensor

fn clone(&self) -> DenseTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DenseTensor

Available on crate feature tensor only.

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

Formats the value using the given formatter.

impl Default for DenseTensor

Available on crate feature tensor only.

fn default() -> Self

Returns the “default value” for a type.

impl From<DenseTensor> for DifferentiableTensor

Convert from DenseTensor

fn from(tensor: DenseTensor) -> Self

Converts to this type from the input type.

impl MessageFunction<DenseTensor> for IdentityMessage

Available on crate feature tensor-gnn only.

fn message( &self, src_features: &DenseTensor, _edge_features: Option<&DenseTensor>, _dst_features: &DenseTensor, ) -> DenseTensor

计算消息

impl MessageFunction<DenseTensor> for LinearMessage

Available on crate feature tensor-gnn only.

fn message( &self, src_features: &DenseTensor, _edge_features: Option<&DenseTensor>, _dst_features: &DenseTensor, ) -> DenseTensor

计算消息

impl PartialEq for DenseTensor

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

impl TensorBase for DenseTensor

Available on crate feature tensor only.

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

获取张量的形状（各维度大小）

fn dtype(&self) -> DType

获取数据类型

fn device(&self) -> Device

获取设备类型

fn to_dense(&self) -> DenseTensor

转换为密集张量

fn to_sparse(&self) -> Option<SparseTensor>

转换为稀疏张量（如果适用）

fn ndim(&self) -> usize

获取张量的维度数（rank）

fn numel(&self) -> usize

获取总元素数量

fn is_scalar(&self) -> bool

检查是否为标量（0 维张量）

fn is_vector(&self) -> bool

检查是否为向量（1 维张量）

fn is_matrix(&self) -> bool

检查是否为矩阵（2 维张量）

impl TensorOps for DenseTensor

Available on crate feature tensor only.

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

张量加法

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

张量减法

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

逐元素乘法（Hadamard 积）

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

逐元素除法

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

矩阵乘法（仅适用于 2D 张量）

fn transpose(&self, axes: Option<&[usize]>) -> Self

转置（交换维度）

fn sum(&self, axes: Option<&[usize]>) -> Self

沿指定轴求和

fn mean(&self, axes: Option<&[usize]>) -> Self

沿指定轴求均值

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

逐元素乘以标量

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

逐元素加上标量

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

where F: Fn(f64) -> f64 + Send + Sync,

逐元素应用函数

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

重塑张量形状

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

切片操作

fn concat(&self, other: &Self, axis: usize) -> Self

拼接两个张量

fn max(&self) -> f64

获取最大值

fn min(&self) -> f64

获取最小值

fn norm(&self) -> f64

获取 L2 范数

fn normalize(&self) -> Self

归一化到单位范数