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 MaxAggregator

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 SumAggregator

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 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

归一化到单位范数

impl StructuralPartialEq for DenseTensor