Struct VirtualTensor

pub struct VirtualTensor<E: Numeric, IO = Read> { /* private fields */ }

Tensor representation that is decoupled from how the tensor is stored.

Implementations

impl<E: Numeric, IO: Clone> VirtualTensor<E, IO>

pub fn as_tensor_map(&self) -> TensorMap<E>

pub fn as_slice(&self, start: u32, end: u32) -> Slice<Line<E>>

pub fn shape(&self, axis: u32) -> u32

Get the shape of the tensor at the given axis.

pub fn stride(&self, axis: u32) -> u32

Get the stride of the tensor at the given axis.

pub fn rank(&self) -> u32

Get the rank of the tensor.

pub fn len(&self) -> u32

pub fn buffer_len(&self) -> u32

pub fn __expand_as_tensor_map( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <TensorMap<E> as CubeType>::ExpandType

pub fn __expand_as_slice( context: &mut Scope, this: <Self as CubeType>::ExpandType, start: <u32 as CubeType>::ExpandType, end: <u32 as CubeType>::ExpandType, ) -> <Slice<Line<E>> as CubeType>::ExpandType

pub fn __expand_shape( scope: &mut Scope, this: <Self as CubeType>::ExpandType, axis: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_stride( scope: &mut Scope, this: <Self as CubeType>::ExpandType, axis: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_rank( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_len( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_buffer_len( scope: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

impl<E: Numeric, IO: Clone> VirtualTensor<E, IO>

pub fn coordinate(&self, index: u32, dim: u32) -> u32

pub fn __expand_coordinate( scope: &mut Scope, this: <Self as CubeType>::ExpandType, index: <u32 as CubeType>::ExpandType, dim: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

impl<E: Numeric> VirtualTensor<E, Read>

pub fn new<V: VirtualTensorOperations<E> + 'static>(_v: &V) -> Self

Create a new read only virtual tensor.

pub fn __expand_new<V>( _scope: &mut Scope, v: V::ExpandType, ) -> VirtualTensorExpand<E, Read>

where V::ExpandType: VirtualTensorOperationsExpand<E>, V: VirtualTensorOperations<E> + CubeType + 'static,

Expand function of Self::new.

impl<E: Numeric> VirtualTensor<E, ReadWrite>

pub fn new<V: VirtualTensorOperations<E> + 'static>(_v: &mut V) -> Self

Create a new read write virtual tensor.

pub fn __expand_new<V>( _scope: &mut Scope, v: V::ExpandType, ) -> VirtualTensorExpand<E, ReadWrite>

where V::ExpandType: VirtualTensorOperationsExpand<E>, V: VirtualTensorOperations<E> + CubeType + 'static,

Expand function of Self::new.

Trait Implementations

impl<E: Clone + Numeric, IO: Clone> Clone for VirtualTensor<E, IO>

fn clone(&self) -> VirtualTensor<E, IO>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<E: Numeric, IO: Clone> CubeType for VirtualTensor<E, IO>

type ExpandType = VirtualTensorExpand<E, IO>

fn into_mut(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.

impl<E: Numeric, IO: Clone> List<Line<E>> for VirtualTensor<E, IO>

fn __expand_read( scope: &mut Scope, this: VirtualTensorExpand<E, IO>, index: <u32 as CubeType>::ExpandType, ) -> <Line<E> as CubeType>::ExpandType

fn read(&self, index: u32) -> T

fn read_unchecked(&self, index: u32) -> T

fn __expand_read_unchecked( scope: &mut Scope, this: Self::ExpandType, index: ExpandElementTyped<u32>, ) -> <T as CubeType>::ExpandType

impl<E: Numeric> ListMut<Line<E>> for VirtualTensor<E, ReadWrite>

fn __expand_write( scope: &mut Scope, this: VirtualTensorExpand<E, ReadWrite>, index: <u32 as CubeType>::ExpandType, value: <Line<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

fn write(&self, index: u32, value: T)

impl<E: Numeric, IO: Clone> Copy for VirtualTensor<E, IO>