Struct web_rwkv::tensor::Tensor

pub struct Tensor<D: Device, T: Scalar> {
    pub context: Context,
    /* private fields */
}

Fields

context: Context

Implementations

impl<D: Device, T: Scalar> Tensor<D, T>

pub fn len(&self) -> usize

pub fn is_empty(&self) -> bool

pub fn size(&self) -> usize

Size of the tensor in bytes.

pub fn offset(index: usize) -> usize

The offset in bytes for a linear index.

pub fn data(&self) -> &D::Data

impl<T: Scalar, K: Kind> Tensor<Gpu<K>, T>

pub fn load(&self, host: &TensorCpu<'_, T>) -> Result<(), TensorError>

pub fn load_batch( &self, host: &TensorCpu<'_, T>, batch: usize ) -> Result<(), TensorError>

pub fn destroy(self)

impl<'a, T: Scalar> Tensor<Cpu<'a, T>, T>

pub fn map<U: Scalar>(self, f: impl FnMut(&T) -> U) -> TensorCpu<'a, U>

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

Repeat the tensor along a given axis.

pub fn split(self, axis: usize) -> Result<Vec<Self>, TensorError>

Split the tensor along the highest plural axis.

pub fn stack(batches: Vec<Self>) -> Result<Self, TensorError>

Concat a batch of tensors.

pub fn slice( &self, x: impl TensorAxis, y: impl TensorAxis, z: impl TensorAxis, w: impl TensorAxis ) -> Result<TensorCpu<'a, T>, TensorError>

pub fn into_slice( self, x: impl TensorAxis, y: impl TensorAxis, z: impl TensorAxis, w: impl TensorAxis ) -> Result<Self, TensorError>

impl<T: Scalar> Tensor<Gpu<ReadWrite>, T>

pub fn view( &self, x: impl TensorAxis, y: impl TensorAxis, z: impl TensorAxis, w: impl TensorAxis ) -> Result<TensorView<'_, T>, TensorError>

Trait Implementations

impl<D: Device, T: Scalar> Clone for Tensor<D, T>

fn clone(&self) -> Self

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<D: Debug + Device, T: Debug + Scalar> Debug for Tensor<D, T>

where D::Data: Debug,

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

Formats the value using the given formatter.

impl<D: Device, T: Scalar> Deref for Tensor<D, T>

type Target = <D as Device>::Data

The resulting type after dereferencing.

fn deref(&self) -> &Self::Target

Dereferences the value.

impl<T: Scalar, K: Kind> From<Tensor<Cpu<'_, T>, T>> for TensorGpu<T, K>

fn from(value: TensorCpu<'_, T>) -> Self

Converts to this type from the input type.

impl<T: Scalar> From<Tensor<Cpu<'_, T>, T>> for Vec<T>

fn from(value: TensorCpu<'_, T>) -> Self

Converts to this type from the input type.

impl<T: Scalar> From<Tensor<Gpu<ReadBack>, T>> for TensorCpu<'_, T>

fn from(value: TensorGpu<T, ReadBack>) -> Self

Converts to this type from the input type.