Enum BackendTensor 

pub enum BackendTensor<B: BackendTypes> {
    Float(B::FloatTensorPrimitive),
    Int(B::IntTensorPrimitive),
    Bool(B::BoolTensorPrimitive),
    Quantized(B::QuantizedTensorPrimitive),
}

Tensor which points to a backend tensor primitive kind.

Variants

Float(B::FloatTensorPrimitive)

Float tensor handle.

Int(B::IntTensorPrimitive)

Int tensor handle.

Bool(B::BoolTensorPrimitive)

Bool tensor handle.

Quantized(B::QuantizedTensorPrimitive)

Quantized tensor handle.

Implementations

impl<B: Backend> BackendTensor<B>

pub fn float(self) -> B::FloatTensorPrimitive

Returns the inner float tensor primitive.

pub fn as_float(&self) -> &B::FloatTensorPrimitive

Returns the inner float tensor primitive.

pub fn int(self) -> B::IntTensorPrimitive

Returns the inner int tensor primitive.

pub fn bool(self) -> B::BoolTensorPrimitive

Returns the inner bool tensor primitive.

pub fn quantized(self) -> B::QuantizedTensorPrimitive

Returns the inner quantized tensor primitive.

pub fn name(&self) -> &'static str

Returns the tensor primitive kind name.

Trait Implementations

impl<B: Clone + BackendTypes> Clone for BackendTensor<B>

where B::FloatTensorPrimitive: Clone, B::IntTensorPrimitive: Clone, B::BoolTensorPrimitive: Clone, B::QuantizedTensorPrimitive: Clone,

fn clone(&self) -> BackendTensor<B>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<B: Debug + BackendTypes> Debug for BackendTensor<B>

where B::FloatTensorPrimitive: Debug, B::IntTensorPrimitive: Debug, B::BoolTensorPrimitive: Debug, B::QuantizedTensorPrimitive: Debug,

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

Formats the value using the given formatter.

impl<B: BackendTypes> TensorMetadata for BackendTensor<B>

type Device = <B as BackendTypes>::Device

The device type associated with the tensor.

fn device(&self) -> Self::Device

Get the device associated with the tensor.

fn dtype(&self) -> DType

Get the dtype of the tensor.

fn shape(&self) -> Shape

Get the shape of the tensor.

fn can_mut(&self) -> bool

Whether the tensor’s buffer can be mutated in place — i.e. this handle uniquely owns it, so an in-place op (slice_assign, an inplace kernel) writes the existing allocation instead of copying it first.

fn rank(&self) -> usize

Get the number of dimensions of the tensor.

fn scheme(&self) -> QuantScheme

Get the quantization scheme for a quantized float tensor.