Struct tension::DeviceTensor

pub struct DeviceTensor<T: Prm + Interop> { /* fields omitted */ }

Tensor structure. It consists of a contiguous one-dimensional array and a shape. Tensor tries to reuse resources as long as possible and implements copy-on-write mechanism.

Implementations

impl<T: Prm + Interop> DeviceTensor<T>

pub unsafe fn new_uninit(loc: &DeviceLocation, shape: &[usize]) -> Self

Create unitialized tensor in the specified location.

pub fn new_filled(loc: &DeviceLocation, shape: &[usize], value: T) -> Self

Create tensor filled with value in the specified location.

pub fn new_zeroed(loc: &DeviceLocation, shape: &[usize]) -> Self

Create tensor filled with zeros in the specified location.

Trait Implementations

impl<T: Prm + Interop> TensorTrait<T> for DeviceTensor<T>

fn shape(&self) -> &[usize]

Shape of the tensor - slice containing all tensor dimensions.

fn reshape(&self, shape: &[usize]) -> Self

Returns a new tensor that shares the same data but has other shape. Failed if the product of all shape dimensions is not equal to buffer size.

fn load(&self, dst: &mut [T])

Load flattened data from tensor to slice.

fn store(&mut self, src: &[T])

Store data from slice to a tensor in a flattened manner.