Struct tensor_compute::GpuTensorView

pub struct GpuTensorView<'a> {
    pub shape_strides: ShapeStrides,
    // some fields omitted
}

A GpuTensorView share the same data as the original Tensor, but can have different shapes and strides For example, the original shape could be [2, 2] and the GpuTensorView could be [1, 2, 2]

Fields

shape_strides: ShapeStrides

Implementations

impl<'a> GpuTensorView<'a>

Used to temporarily modify how the underlying tensor data is interpreted, by changing the tensor shape or strides for example

pub fn from_tensor(gpu_tensor: &'a GpuTensor, dim_strides: ShapeStrides) -> Self

pub async fn to_tensor<'_>(&'_ self) -> GpuTensor

pub fn slice<T: Into<SliceRangeInfo>>(
    &self,
    bounds: Vec<T>
) -> GpuTensorView<'_>

pub fn buffer_size_in_bytes(&self) -> usize

pub fn shape(&self) -> &VecDeque<usize>

pub fn strides(&self) -> &VecDeque<usize>

pub fn dim_strides(&self) -> &ShapeStrides

impl<'a> GpuTensorView<'a>

pub async fn contiguous<'_>(&'_ self) -> GpuTensor

pub async fn eq<'_, '_>(&'_ self, other: &'_ Self) -> bool

Trait Implementations

impl<'a> AsShaderInput for GpuTensorView<'a>

fn to_shader_inputs(&self, binding_offset: usize) -> Vec<ShaderInput<'_>>

impl<'a> GpuAllocated for GpuTensorView<'a>

fn get_gpu(&self) -> &'static GpuInstance

fn internal_gpu_buffer(&self) -> &GpuBuffer

fn internal_buffer_size_in_bytes(&self) -> usize

impl<'a> MutShapeStrideTrait for GpuTensorView<'a>

fn increase_rank(&mut self)

Artificially increase the rank, setting the new dimension shape to 1

fn decrease_rank(&mut self)

Decrease the rank if the leading shape dimension is 1, otherwise panics

impl<'a> ShapeStrideTrait for GpuTensorView<'a>

fn shape(&self) -> &VecDeque<usize>

fn strides(&self) -> &VecDeque<usize>

fn offset(&self) -> usize

fn rank(&self) -> usize

fn numel(&self) -> usize

fn numel_from_shape(shape: &VecDeque<usize>) -> usize