Enum BackendTensor 

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

Tensor which points to a backend tensor primitive kind.

Variants

Float(<B as BackendTypes>::FloatTensorPrimitive)

Float tensor handle.

Int(<B as BackendTypes>::IntTensorPrimitive)

Int tensor handle.

Bool(<B as BackendTypes>::BoolTensorPrimitive)

Bool tensor handle.

Quantized(<B as BackendTypes>::QuantizedTensorPrimitive)

Quantized tensor handle.

Autodiff(<Autodiff<B> as BackendTypes>::FloatTensorPrimitive)

Autodiff float tensor handle.

Implementations

impl<B> BackendTensor<B>

where B: Backend,

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

Returns the inner float tensor primitive.

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

Returns the inner float tensor primitive.

pub fn int(self) -> <B as BackendTypes>::IntTensorPrimitive

Returns the inner int tensor primitive.

pub fn bool(self) -> <B as BackendTypes>::BoolTensorPrimitive

Returns the inner bool tensor primitive.

pub fn quantized(self) -> <B as BackendTypes>::QuantizedTensorPrimitive

Returns the inner quantized tensor primitive.

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

Returns the inner autodiff tensor primitive.

pub fn as_autodiff( &self, ) -> &<Autodiff<B> as BackendTypes>::FloatTensorPrimitive

Returns the inner autodiff tensor primitive.

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

Returns the inner autodiff tensor primitive.

pub fn into_autodiff(self) -> BackendTensor<Autodiff<B>>

Lift a handle for backend B into the equivalent handle for Autodiff<B>.

An already-tracked float (Autodiff) becomes the Float handle of Autodiff<B>; int/bool/ quantized handles are re-tagged unchanged (those primitives are shared between B and Autodiff<B>). An untracked Float handle is invalid here (under autodiff, float tensors arrive tracked), so it panics.

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

Returns the tensor primitive kind name.

Trait Implementations

impl<B> Clone for BackendTensor<B>

where B: Clone + BackendTypes, <B as BackendTypes>::FloatTensorPrimitive: Clone, <B as BackendTypes>::IntTensorPrimitive: Clone, <B as BackendTypes>::BoolTensorPrimitive: Clone, <B as BackendTypes>::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 for BackendTensor<B>

where B: Debug + BackendTypes, <B as BackendTypes>::FloatTensorPrimitive: Debug, <B as BackendTypes>::IntTensorPrimitive: Debug, <B as BackendTypes>::BoolTensorPrimitive: 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 BackendTensor<B>

where B: BackendTypes,

type Device = <B as BackendTypes>::Device

The device type associated with the tensor.

fn device(&self) -> <BackendTensor<B> as TensorMetadata>::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.