Struct TensorBuffer 

pub struct TensorBuffer<B: Backend, T: TensorElement> { /* private fields */ }

A typed, one-dimensional collective buffer on a tensor backend.

Clones refer to the same collective state. Exported tensor primitives are snapshots: later collective writes use the backend’s copy-on-write semantics and do not modify previously exported tensors.

Implementations

impl<B: Backend, T: TensorElement> TensorBuffer<B, T>

pub fn len(&self) -> usize

pub fn is_empty(&self) -> bool

pub fn dtype(&self) -> DType

pub fn device(&self) -> &B::Device

pub fn float_tensor(&self) -> Result<B::FloatTensorPrimitive, TensorDeviceError>

Export the current floating tensor without a host transfer.

pub fn int_tensor(&self) -> Result<B::IntTensorPrimitive, TensorDeviceError>

Export the current integer tensor without a host transfer.

Trait Implementations

impl<B: Clone + Backend, T: Clone + TensorElement> Clone for TensorBuffer<B, T>

where B::Device: Clone,

fn clone(&self) -> TensorBuffer<B, T>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<B: Debug + Backend, T: Debug + TensorElement> Debug for TensorBuffer<B, T>

where B::Device: Debug,

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

Formats the value using the given formatter.