Struct PooledTensor 

pub struct PooledTensor<'pool> { /* private fields */ }

池化张量：带有自动回收功能的张量包装器

Implementations

impl<'pool> PooledTensor<'pool>

pub fn tensor(&self) -> &DenseTensor

获取内部张量引用

pub fn tensor_mut(&mut self) -> &mut DenseTensor

获取内部张量可变引用

pub fn into_inner(self) -> DenseTensor

消耗包装器并返回内部张量（不回收）

Methods from Deref<Target = DenseTensor>

pub fn nbytes(&self) -> usize

获取字节大小

pub fn is_contiguous(&self) -> bool

检查是否连续存储

pub fn alignment(&self) -> usize

获取对齐字节数

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>

获取指定列的数据

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 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<'pool> Clone for PooledTensor<'pool>

Available on crate feature tensor-pool only.

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'pool> Deref for PooledTensor<'pool>

Available on crate feature tensor-pool only.

type Target = DenseTensor

The resulting type after dereferencing.

fn deref(&self) -> &Self::Target

Dereferences the value.

impl<'pool> DerefMut for PooledTensor<'pool>

Available on crate feature tensor-pool only.

fn deref_mut(&mut self) -> &mut Self::Target

Mutably dereferences the value.

impl<'pool> Drop for PooledTensor<'pool>

Available on crate feature tensor-pool only.

fn drop(&mut self)

Executes the destructor for this type.

impl<'pool> Send for PooledTensor<'pool>

Available on crate feature tensor-pool only.

Safety

PooledTensor 可以安全地发送到其他线程，因为：

1. 内部 tensor: DenseTensor 已实现 Send
2. pool 指针仅用于在 Drop 时回收张量，不在线程间共享状态
3. 生命周期 'pool 确保指针有效性

impl<'pool> Sync for PooledTensor<'pool>

Available on crate feature tensor-pool only.

Safety

PooledTensor 可以安全地在线程间共享，因为：

1. 内部 tensor: DenseTensor 已实现 Sync
2. pool 指针仅在 Drop 时访问，且 TensorPool 本身是 Sync 的
3. 所有可变操作都通过 &mut self 方法（如 tensor_mut()）进行，由 Rust 借用规则保证安全