Struct DispatchTensor 

pub struct DispatchTensor {
    pub kind: DispatchTensorKind,
    pub autodiff: DispatchAutodiffContext,
}

A tensor that can dispatch operations to any enabled backend at runtime.

Fields

kind: DispatchTensorKind

Tensor kind primitive.

autodiff: DispatchAutodiffContext

Autodiff backend association for this tensor.

Trait Implementations

impl Clone for DispatchTensor

fn clone(&self) -> DispatchTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DispatchTensor

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

Formats the value using the given formatter.

impl<C> DispatchKindConversion<Autodiff<Flex, C>> for DispatchTensor

where C: CheckpointStrategy + IntoGradientCheckpointingStrategy,

Available on crate features autodiff and flex only.

fn try_into_backend( tensor: DispatchTensor, ) -> Result<BackendTensor<Autodiff<Flex, C>>, String>

Attempts to extract a backend-specific BackendTensor wrapper from a generic, dynamically-routed DispatchTensor.

fn from_backend(tensor: BackendTensor<Autodiff<Flex, C>>) -> DispatchTensor

Encapsulates a backend-specific tensor variant back into a globally routing DispatchTensor.

impl DispatchKindConversion<Flex> for DispatchTensor

Available on crate feature flex only.

fn try_into_backend( tensor: DispatchTensor, ) -> Result<BackendTensor<Flex>, String>

Attempts to extract a backend-specific BackendTensor wrapper from a generic, dynamically-routed DispatchTensor.

fn from_backend(tensor: BackendTensor<Flex>) -> DispatchTensor

Encapsulates a backend-specific tensor variant back into a globally routing DispatchTensor.

impl From<BridgeTensor> for DispatchTensor

fn from(value: BridgeTensor) -> DispatchTensor

Converts to this type from the input type.

impl TensorMetadata for DispatchTensor

type Device = DispatchDevice

The device type 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 device(&self) -> <DispatchTensor as TensorMetadata>::Device

Get the device associated with the tensor.

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.