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".

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