Struct tension::CommonTensor

pub struct CommonTensor<T: Prm, Buf: Buffer<T>> {
    pub buffer: Rc<Buf>,
    pub shape: Shape,
    // some fields omitted
}

An intermediate structure that contains most of the Tensor functionality.

Fields

buffer: Rc<Buf>

shape: Shape

Implementations

impl<T: Prm, Buf: Buffer<T>> CommonTensor<T, Buf>

pub fn from_shared_buffer(rc_buffer: Rc<Buf>, shape: &Shape) -> Self

Create tensor from shared buffer and shape

pub fn from_buffer(buffer: Buf, shape: &Shape) -> Self

Create tensor from specified buffer and shape

pub fn buffer(&self) -> &Buf

Provides access to inner buffer.

pub fn buffer_mut(&mut self) -> &mut Buf

Clones inner buffer if it is shared and provides mutable access to it.

Trait Implementations

impl<T: Prm, Buf: Buffer<T>> Tensor<T> for CommonTensor<T, Buf>

type Buffer = Buf

Inner buffer type.

unsafe fn new_uninit_in(context: &Buf::Context, shape: &Shape) -> Self

Create unitialized tensor

fn new_filled_in(context: &Buf::Context, shape: &Shape, value: T) -> Self

Create tensor filled with value on the specified hardware

fn new_zeroed_in(context: &Buf::Context, shape: &Shape) -> Self

Create tensor filled with zeros on the specified hardware

fn shape(&self) -> &Shape

Shape of the tensor - a slice containing all tensor dimensions.

fn reshape(&self, shape: &Shape) -> 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.