Struct TensorView

pub struct TensorView<'a> { /* private fields */ }

A view into a tensor’s data

Implementations

impl<'a> TensorView<'a>

pub fn new(tensor: &'a Tensor) -> Self

Create a new tensor view

pub fn new_with_params( tensor: &'a Tensor, offset: usize, shape: Vec<usize>, strides: Vec<usize>, ) -> Result<Self>

Create a view with custom offset, shape, and strides

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

Get the shape of this view

pub fn strides(&self) -> &[usize]

Get the strides of this view

pub fn offset(&self) -> usize

Get the offset of this view

pub fn ndim(&self) -> usize

Get the number of dimensions

pub fn size(&self) -> usize

Get the total number of elements in this view

pub fn tensor(&self) -> &Tensor

Get the underlying tensor

pub fn is_contiguous(&self) -> bool

Check if this view is contiguous in memory

pub fn slice(&self, dim: usize, range: Range<usize>) -> Result<TensorView<'a>>

Create a slice of this view along a specific dimension

pub fn select(&self, dim: usize, index: usize) -> Result<TensorView<'a>>

Create a view of a specific index along a dimension (reduces dimensionality by 1)

pub fn transpose(&self) -> Result<TensorView<'a>>

Transpose the view (swap last two dimensions)

pub fn reshape(&self, new_shape: &[usize]) -> Result<TensorView<'a>>

Reshape the view (only works if contiguous)

pub fn squeeze(&self) -> TensorView<'a>

Create a view with squeezed dimensions (remove dimensions of size 1)

pub fn unsqueeze(&self, dim: usize) -> Result<TensorView<'a>>

Create a view with an added dimension of size 1

pub fn unravel_index(&self, index: usize) -> Result<Vec<usize>>

Convert linear index to multi-dimensional indices

pub fn ravel_index(&self, indices: &[usize]) -> Result<usize>

Convert multi-dimensional indices to linear index in the view

pub fn get(&self, indices: &[usize]) -> Result<f32>

Get a scalar value at the given indices

pub fn to_tensor(&self) -> Result<Tensor>

Convert view to a new tensor (copies data)

pub fn iter(&'a self) -> TensorViewIterator<'a>

Iterate over all elements in the view

pub fn axis_iter(&'a self, axis: usize) -> Result<AxisIterator<'a>>

Create a view that iterates along a specific axis

Trait Implementations

impl<'a> Clone for TensorView<'a>

fn clone(&self) -> TensorView<'a>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'a> Debug for TensorView<'a>

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

Formats the value using the given formatter.

impl<'a> Index<usize> for TensorView<'a>

type Output = f32

The returned type after indexing.

fn index(&self, index: usize) -> &Self::Output

Performs the indexing (container[index]) operation.