Struct HostTensor 

pub struct HostTensor<D: Scalar, Element: M> { /* private fields */ }

Tensor stored in host memory.

Implementations

impl<D: Scalar, Element: M> HostTensor<D, Element>

pub type Mapping = Element

Mapping type alias.

pub fn from_buf(data: Vec<Opt<D>>) -> Self

Creates a tensor from a vector.

pub fn from_safetensors(view: &TensorView<'_>) -> Result<Self, SafeTensorError>

where D: ScalarBytes,

Creates a tensor from a safetensors tensor view.

The view’s per-axis shape must match Element’s pair-flattened size list (e.g. m![H, X] expects safetensors shape [H.size, X.size]) and its bytes are decoded as little-endian D values — LE is mandated by the safetensors format spec, not our choice. Returns [safetensors::SafeTensorError::TensorInvalidInfo] on any mismatch.

pub fn zero() -> Self

Creates a tensor filled with zeros.

pub fn rand(rng: &mut impl Rng) -> Self

where StandardUniform: Distribution<D>,

Creates a tensor filled with random values.

pub fn uninit() -> Self

Creates an uninitialized tensor.

pub async fn to_hbm<Chip: M, Element2: M>( &self, _dma: &mut DmaContext<{ Dma::Pcie }>, address: Address, ) -> HbmTensor<D, Chip, Element2>

Converts to HBM tensor.

TODO: address should be optional.

pub fn into_inner(self) -> Tensor<D, Self::Mapping>

Consumes self and returns the inner tensor value.

pub fn to_buf(&self) -> Vec<Opt<D>>

Returns the tensor data as a flat vector in physical layout order.

Trait Implementations

impl<D: Clone + Scalar, Element: Clone + M> Clone for HostTensor<D, Element>

fn clone(&self) -> HostTensor<D, Element>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<D: Debug + Scalar, Element: Debug + M> Debug for HostTensor<D, Element>

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

Formats the value using the given formatter.

impl<D: Scalar, Element: M> From<Tensor<D, Element>> for HostTensor<D, Element>

fn from(inner: Tensor<D, Element>) -> Self

Converts to this type from the input type.