Enum burn_tensor::TensorPrimitive

pub enum TensorPrimitive<B: Backend, const D: usize> {
    Float(B::FloatTensorPrimitive<D>),
    QFloat(B::QuantizedTensorPrimitive<D>),
}

A primitive tensor representation.

Variants

Float(B::FloatTensorPrimitive<D>)

Float tensor primitive.

QFloat(B::QuantizedTensorPrimitive<D>)

Quantized float tensor primitive.

Implementations

impl<B: Backend, const D: usize> TensorPrimitive<B, D>

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

Returns the full tensor representation.

Trait Implementations

impl<B: Clone + Backend, const D: usize> Clone for TensorPrimitive<B, D>

where B::FloatTensorPrimitive<D>: Clone, B::QuantizedTensorPrimitive<D>: Clone,

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

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<B: Debug + Backend, const D: usize> Debug for TensorPrimitive<B, D>

where B::FloatTensorPrimitive<D>: Debug, B::QuantizedTensorPrimitive<D>: Debug,

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

Formats the value using the given formatter.