Struct PooledTensor 

pub struct PooledTensor<T: TensorElement + Default> { /* private fields */ }

A pooled tensor that automatically returns memory to pool when dropped

Implementations

impl<T: TensorElement + Copy + Default> PooledTensor<T>

pub fn new(shape: &[usize], device: DeviceType) -> Result<Self>

Create a new pooled tensor

pub fn zeros(shape: &[usize], device: DeviceType) -> Result<Self>

Create pooled zeros tensor

pub fn ones(shape: &[usize], device: DeviceType) -> Result<Self>

where T: Add<Output = T> + From<f32>,

Create pooled ones tensor

pub fn tensor(&self) -> &Tensor<T>

Get reference to the underlying tensor

pub fn tensor_mut(&mut self) -> &mut Tensor<T>

Get mutable reference to the underlying tensor

pub fn into_tensor(self) -> Tensor<T>

Convert to owned tensor (removes from pool management)

Trait Implementations

impl<T: Debug + TensorElement + Default> Debug for PooledTensor<T>

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

Formats the value using the given formatter.

impl<T: TensorElement + Default> Drop for PooledTensor<T>

fn drop(&mut self)

Executes the destructor for this type.