Struct TensorHandle 

pub struct TensorHandle<R>
where
    R: Runtime,
{
    pub handle: Handle,
    pub shape: Vec<usize>,
    pub strides: Vec<usize>,
    pub dtype: StorageType,
    /* private fields */
}

Tensor representation containing a server handle as well as basic tensor metadata.,

Fields

handle: Handle

The buffer where the data are stored.

shape: Vec<usize>

strides: Vec<usize>

dtype: StorageType

The type used as storage.

Implementations

impl<R> TensorHandle<R>

where R: Runtime,

pub fn new( handle: Handle, shape: Vec<usize>, strides: Vec<usize>, storage: StorageType, ) -> Self

Create a new tensor.

pub fn empty( client: &ComputeClient<R::Server>, shape: Vec<usize>, storage: StorageType, ) -> Self

pub fn from_ref(handle: &TensorHandleRef<'_, R>, storage: StorageType) -> Self

Create a new tensor.

pub fn new_contiguous( shape: Vec<usize>, handle: Handle, storage: StorageType, ) -> Self

Create a new tensor with a contiguous memory layout.

pub fn can_mut(&self) -> bool

Check if the tensor is safe to mutate.

pub fn as_ref(&self) -> TensorHandleRef<'_, R>

pub fn as_arg<'a>(&'a self, vectorisation: u8) -> TensorArg<'a, R>

Return the reference to a tensor argument.

pub fn as_copy_descriptor<'a>(&'a self) -> CopyDescriptor<'a>

impl<R> TensorHandle<R>

where R: Runtime,

pub fn zeros( client: &ComputeClient<R::Server>, shape: Vec<usize>, dtype: StorageType, ) -> Self

Trait Implementations

impl<R> Clone for TensorHandle<R>

where R: Runtime,

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<R> Debug for TensorHandle<R>

where R: Runtime,

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

Formats the value using the given formatter.