Struct Tensor 

pub struct Tensor { /* private fields */ }

A contiguous, row-major tensor on a specific device.

Implementations

impl Tensor

pub fn from_f32( data: &[f32], shape: impl Into<Shape>, device: &Device, ) -> Result<Tensor>

pub fn from_u32( data: &[u32], shape: impl Into<Shape>, device: &Device, ) -> Result<Tensor>

pub fn zeros(shape: impl Into<Shape>, device: &Device) -> Result<Tensor>

pub fn shape(&self) -> &Shape

pub fn dtype(&self) -> DType

pub fn device(&self) -> Device

pub fn to_vec_f32(&self) -> Result<Vec<f32>>

Copy the tensor contents to a host Vec<f32>. Sync facade: on wasm32 only CPU tensors can be read synchronously (roadmap v4, pitfall 14) — use Tensor::to_vec_f32_async for WebGPU tensors there.

pub fn to_vec_u32(&self) -> Result<Vec<u32>>

pub async fn to_vec_f32_async(&self) -> Result<Vec<f32>>

Async readback — the primary form (identical result to to_vec_f32); on wasm the browser event loop drives the buffer mapping.

pub async fn to_vec_f32_batch(tensors: &[Tensor]) -> Result<Vec<Vec<f32>>>

Read several f32 tensors back in one GPU round trip, in the order given.

Tensor::to_vec_f32_async is a submit and a fence wait per call, so a step that wants several small tensors — the attention probe reads n_layer + 1 per generated token — pays for the round trips, not the bytes. Every tensor must be on the same device.

pub async fn to_vec_u32_async(&self) -> Result<Vec<u32>>

pub fn reshape(&self, shape: impl Into<Shape>) -> Result<Tensor>

Reinterpret the shape; element count must be unchanged. Zero-copy.

pub fn narrow_rows(&self, start: usize, count: usize) -> Result<Tensor>

View of count rows of a rank-2 tensor starting at start. Zero-copy on WebGPU (subject to 256-byte offset alignment), copied on CPU.

Trait Implementations

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Tensor

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

Formats the value using the given formatter.