Struct PtrTensor

pub struct PtrTensor<P> { /* private fields */ }

A generic Tensor representation.

Implementations

impl PtrTensor<DevicePtr>

pub fn alloc_simple(device: CudaDevice, shape: Vec<usize>, dtype: DType) -> Self

pub fn device(&self) -> CudaDevice

pub fn deep_clone(&self) -> DeviceTensor

pub unsafe fn copy_simple_from_host(&self, buffer: &[u8])

pub unsafe fn copy_simple_to_host(&self, buffer: &mut [u8])

pub fn copy_from_as_scalar_op( &self, other: &DeviceTensor, ) -> ScalarOpArgs<DevicePtr>

pub unsafe fn copy_from(&self, other: &DeviceTensor)

pub unsafe fn copy_from_host_staged(&self, buffer: &[u8])

A (potentially) slower version of Self::copy_simple_from_host that works for any strides, by potentially copying to an intermediate stage on the device.

pub unsafe fn copy_to_host_staged(&self, buffer: &mut [u8])

A (potentially) slower version of Self::copy_simple_to_host that works for any strides, by potentially copying to an intermediate stage on the device.

impl<P> PtrTensor<P>

pub fn from_parts(ptr: P, shape: StridedShape, dtype: DType) -> Self

pub fn into_ptr(self) -> P

pub fn ptr(&self) -> &P

pub fn strided_shape(&self) -> &StridedShape

pub fn dtype(&self) -> DType

pub fn dense_size_bytes(&self) -> usize

pub fn map_ptr<K>(self, f: impl FnOnce(P) -> K) -> PtrTensor<K>

impl<P: OffsetPtr> PtrTensor<P>

pub fn permute(&self, permutation: &[usize]) -> Self

pub fn view(&self, new_shape: Vec<usize>) -> Result<Self, ViewError>

pub fn broadcast(&self, new_shape: Vec<usize>) -> Self

pub fn slice(&self, axis: usize, range: impl Into<SliceRange>) -> Self

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

pub fn flip(&self, axis: usize) -> Self

pub fn repeat_unary(&self, axis: usize, count: usize) -> Self

impl<P: Clone> PtrTensor<P>

pub fn to_mat_mul_arg(&self) -> MatMulOperand<P>

Trait Implementations

impl<P: Clone> Clone for PtrTensor<P>

fn clone(&self) -> PtrTensor<P>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<P: Debug> Debug for PtrTensor<P>

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

Formats the value using the given formatter.

impl<P: PartialEq> PartialEq for PtrTensor<P>

fn eq(&self, other: &PtrTensor<P>) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl<P: Eq> Eq for PtrTensor<P>

impl<P> StructuralPartialEq for PtrTensor<P>