Struct Storage 

pub struct Storage { /* private fields */ }

An owned, reference-counted buffer of elements on one device.

Multiple tensors (views) may share one storage; storage never knows how many.

Implementations

impl Storage

pub fn cpu_f32_zeros(numel: usize) -> Self

Zero-initialized CPU storage for numel f32 elements.

pub fn from_f32_vec(data: Vec<f32>) -> Self

CPU storage owning data as f32 elements.

pub fn from_f64_vec(data: Vec<f64>) -> Self

CPU storage owning data as f64 elements.

pub fn from_i64_vec(data: Vec<i64>) -> Self

CPU storage owning data as i64 elements.

pub fn dtype(&self) -> DType

The element type of this buffer.

pub fn device(&self) -> Device

The device this buffer lives on.

pub fn data(&self) -> &StorageData

The raw storage data.

pub fn cpu(&self) -> Result<&CpuStorage>

The typed CPU buffer, or a typed error when the storage is on a different device.

pub fn from_opaque(buffer: OpaqueBuffer, dtype: DType, device: Device) -> Self

Storage backed by an out-of-tree backend’s allocation.

pub fn opaque(&self) -> Option<&OpaqueBuffer>

The opaque buffer, when this storage lives on an out-of-tree backend.

Trait Implementations

impl Debug for Storage

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

Formats the value using the given formatter.