Struct Tensor 

pub struct Tensor { /* private fields */ }

GPU tensor with shape metadata. Wraps a GpuBuffer. Shape is stored inline (max 6 dims covers batch x channel x D x H x W + extra).

Implementations

impl Tensor

pub fn new(dev: &GpuDevice, data: &[f32], shape: &[u32]) -> Result<Self>

Create a tensor from data with the given shape.

pub fn from_buf(buf: GpuBuffer, shape: &[u32]) -> Result<Self>

Create a tensor from an existing GpuBuffer with the given shape.

pub fn zeros(dev: &GpuDevice, shape: &[u32]) -> Result<Self>

Create a zero tensor with the given shape.

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

Shape as a slice.

pub fn ndim(&self) -> usize

Number of dimensions.

pub fn numel(&self) -> usize

Total number of elements.

pub fn to_vec(&self, dev: &GpuDevice) -> Result<Vec<f32>>

Read tensor data back to CPU.

pub fn buffer(&self) -> &GpuBuffer

Borrow the underlying GpuBuffer.

pub fn reshape(self, new_shape: &[u32]) -> Result<Self>

Reshape to a new shape (same total elements, no data copy).

pub fn dim(&self, i: usize) -> u32

Get a single dimension size.