Struct TensorStorage 

pub struct TensorStorage { /* private fields */ }

Reference-counted tensor storage for memory management

Implementations

impl TensorStorage

pub fn new(data: Vec<f32>, shape: Vec<usize>, dtype: DType) -> Self

Create new tensor storage with data

pub fn new_pooled(shape: Vec<usize>, dtype: DType) -> FfiResult<Self>

Create new tensor storage with pooled allocation

pub fn from_external(data: Vec<f32>, shape: Vec<usize>, dtype: DType) -> Self

Create tensor storage from external data (e.g., NumPy array)

pub fn data(&self) -> MutexGuard<'_, Vec<f32>>

Get a reference to the data (read-only)

pub fn data_mut(&self) -> MutexGuard<'_, Vec<f32>>

Get a mutable reference to the data

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

Get shape

pub fn dtype(&self) -> DType

Get dtype

pub fn view(&self, new_shape: Vec<usize>) -> FfiResult<Self>

Create a view with different shape (same data)

pub fn ref_count(&self) -> usize

Get reference count

pub fn is_external(&self) -> bool

Check if this storage comes from external memory

pub fn grad(&self) -> MutexGuard<'_, Option<Vec<f32>>>

Get gradient data (None if no gradient computed)

pub fn set_grad(&self, grad_data: Option<Vec<f32>>)

Set gradient data

pub fn clear_grad(&self)

Clear gradient

pub fn version(&self) -> usize

Get current version (for gradient tracking) Uses atomic load for lock-free read access

pub fn increment_version(&self)

Increment version (called on in-place operations) Uses atomic increment for lock-free update

Trait Implementations

impl Clone for TensorStorage

fn clone(&self) -> TensorStorage

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorStorage

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

Formats the value using the given formatter.

impl Drop for TensorStorage

fn drop(&mut self)

Executes the destructor for this type.

fn pin_drop(self: Pin<&mut Self>)

🔬This is a nightly-only experimental API. (pin_ergonomics)

Execute the destructor for this type, but different to Drop::drop, it requires self to be pinned.