Struct TensorView

pub struct TensorView<'a, T, const N: usize, A: TensorAllocator> {
    pub storage: &'a TensorStorage<T, A>,
    pub shape: [usize; N],
    pub strides: [usize; N],
}

A view into a tensor.

Fields

storage: &'a TensorStorage<T, A>

Reference to the storage held by the another tensor.

shape: [usize; N]

The shape of the tensor.

strides: [usize; N]

The strides of the tensor.

Implementations

impl<'a, T, const N: usize, A: TensorAllocator + 'static> TensorView<'a, T, N, A>

pub fn as_slice(&self) -> &[T]

Returns the data slice of the tensor.

pub fn as_ptr(&self) -> *const T

Returns the data pointer of the tensor.

pub fn numel(&self) -> usize

Returns the length of the tensor.

pub fn get_unchecked(&self, index: [usize; N]) -> &T

Get the element at the given index.

Returns

A reference to the element at the given index.

Safety

The caller must ensure that the index is within the bounds of the tensor.

pub fn as_contiguous(&self) -> Tensor<T, N, CpuAllocator>

where T: Clone,

Convert the view an owned tensor with contiguous memory.

Returns

A new Tensor instance with contiguous memory.