Struct burn_wgpu::tensor::WgpuTensor

pub struct WgpuTensor<E: WgpuElement, const D: usize> {
    pub client: WgpuComputeClient,
    pub handle: WgpuHandle,
    pub shape: Shape<D>,
    pub device: WgpuDevice,
    pub strides: [usize; D],
    /* private fields */
}

The basic tensor primitive struct.

Fields

client: WgpuComputeClient

Compute client for wgpu.

handle: WgpuHandle

The buffer where the data are stored.

shape: Shape<D>

The shape of the current tensor.

device: WgpuDevice

The device of the current tensor.

strides: [usize; D]

The strides of the current tensor.

Implementations

impl<E: WgpuElement, const D: usize> WgpuTensor<E, D>

pub fn new( client: WgpuComputeClient, device: WgpuDevice, shape: Shape<D>, handle: WgpuHandle ) -> Self

Create a new tensor.

pub fn to_client(&self, client: WgpuComputeClient, device: WgpuDevice) -> Self

Change the context of the current tensor and return the newly transferred tensor.

pub fn copy(&self) -> Self

Copy the current tensor.

pub fn can_mut(&self) -> bool

Check if the tensor is safe to mutate.

pub fn assert_is_on_same_device(&self, other: &Self)

Assert that both tensors are on the same device.

pub fn is_contiguous(&self) -> bool

Check if the current tensor is contiguous.

Trait Implementations

impl<E: Clone + WgpuElement, const D: usize> Clone for WgpuTensor<E, D>

fn clone(&self) -> WgpuTensor<E, D>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<E: Debug + WgpuElement, const D: usize> Debug for WgpuTensor<E, D>

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

Formats the value using the given formatter.