Trait Cast 

pub trait Cast<B, K>
where
    B: Backend,
    K: TensorKind<B>,
{
    type OutputKind: TensorKind<B>;

    // Required method
    fn cast(
        primitive: <K as TensorKind<B>>::Primitive,
        dtype: Self,
    ) -> <Self::OutputKind as TensorKind<B>>::Primitive;
}

Trait for types that represent a valid cast target from a tensor of kind K.

The generic parameter K is the input tensor kind (Float, Int, or Bool). Implementors declare the output kind and provide the actual cast logic.

Required Associated Types

type OutputKind: TensorKind<B>

The output tensor kind after casting.

Required Methods

fn cast( primitive: <K as TensorKind<B>>::Primitive, dtype: Self, ) -> <Self::OutputKind as TensorKind<B>>::Primitive

Cast a tensor primitive to the target dtype.

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<B> Cast<B, Bool> for FloatDType

where B: Backend,

type OutputKind = Float

impl<B> Cast<B, Bool> for IntDType

where B: Backend,

type OutputKind = Int

impl<B> Cast<B, Float> for DType

where B: Backend,

Backward-compatible impl: only float DType variants are accepted.

Panics

Panics if dtype is not a float variant (e.g., DType::I32). Use IntDType directly for cross-kind casting to int.

type OutputKind = Float

impl<B> Cast<B, Float> for FloatDType

where B: Backend,

type OutputKind = Float

impl<B> Cast<B, Float> for IntDType

where B: Backend,

type OutputKind = Int

impl<B> Cast<B, Int> for DType

where B: Backend,

Backward-compatible impl: only int DType variants are accepted.

Panics

Panics if dtype is not an int variant (e.g., DType::F32). Use FloatDType directly for cross-kind casting to float.

type OutputKind = Int

impl<B> Cast<B, Int> for FloatDType

where B: Backend,

type OutputKind = Float

impl<B> Cast<B, Int> for IntDType

where B: Backend,

type OutputKind = Int