Struct TensorBuilder

pub struct TensorBuilder<const DIM: usize> { /* private fields */ }

Helper struct for creating gpu storage buffers (scalars, vectors, matrices, tensors).

When building a scalar, vector, or matrix tensor, it might be more convenient to call GpuScalar::init, GpuVector::init, GpuMatrix::init (or their encase variants: GpuScalar::encase, GpuVector::encase, GpuMatrix::encase; or their uninitialized variants GpuScalar::uninit, GpuVector::uninit, GpuMatrix::uninit).

Implementations

impl TensorBuilder<0>

pub fn scalar(usage: BufferUsages) -> Self

Starts building a storage buffer containing a single scalar value.

impl TensorBuilder<1>

pub fn vector(dim: u32, usage: BufferUsages) -> Self

Starts building a storage buffer containing a vector.

impl TensorBuilder<2>

pub fn matrix(nrows: u32, ncols: u32, usage: BufferUsages) -> Self

Starts building a storage buffer containing a single matrix with nrows rows and ncols columns.

impl<const DIM: usize> TensorBuilder<DIM>

pub fn tensor(shape: [u32; DIM], usage: BufferUsages) -> Self

Starts building a storage buffer containing a tensor with the specified shape.

pub fn label(self, label: String) -> Self

Sets the debug label of this tensor.

pub fn build<T: Pod>(self, device: &Device) -> GpuTensor<T, DIM>

Builds the gpu tensor.

pub fn build_bytes<T>(self, device: &Device, data: &[u8]) -> GpuTensor<T, DIM>

Builds this tensor with raw bytes given for its initial value.

pub fn build_encase<T>( self, device: &Device, data: impl AsRef<[T]>, ) -> GpuTensor<T, DIM>

where T: ShaderType + ShaderSize + WriteInto,

Builds this tensor with raw bytes given for its initial value.

pub fn build_init<T: Pod>( self, device: &Device, data: &[T], ) -> GpuTensor<T, DIM>

Builds this tensor with an array of values given for its initial value.