Enum TensorPrimitive 

pub enum TensorPrimitive<B>
where
    B: Backend,
{
    Float(<B as BackendTypes>::FloatTensorPrimitive),
    QFloat(<B as BackendTypes>::QuantizedTensorPrimitive),
}

A primitive tensor representation.

Variants

Float(<B as BackendTypes>::FloatTensorPrimitive)

Float tensor primitive.

QFloat(<B as BackendTypes>::QuantizedTensorPrimitive)

Quantized float tensor primitive.

Implementations

impl<B> TensorPrimitive<B>

where B: Backend,

pub fn tensor(self) -> <B as BackendTypes>::FloatTensorPrimitive

Returns the full tensor representation.

pub fn get_mut_ref(&mut self) -> &mut <B as BackendTypes>::FloatTensorPrimitive

Returns a mutable reference to the full tensor representation.

Trait Implementations

impl<B> Clone for TensorPrimitive<B>

where B: Clone + Backend, <B as BackendTypes>::FloatTensorPrimitive: Clone, <B as BackendTypes>::QuantizedTensorPrimitive: Clone,

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

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<B> Debug for TensorPrimitive<B>

where B: Debug + Backend, <B as BackendTypes>::FloatTensorPrimitive: Debug, <B as BackendTypes>::QuantizedTensorPrimitive: Debug,

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

Formats the value using the given formatter.

impl<B> TensorMetadata for TensorPrimitive<B>

where B: Backend,

fn dtype(&self) -> DType

The dtype of the tensor.

fn shape(&self) -> Shape

The shape of the tensor.

fn rank(&self) -> usize

The number of dimensions of the tensor.