Trait burn::tensor::TensorKind

pub trait TensorKind<B>: Clone + Debug
where
    B: Backend,
{
    type Primitive: Clone + Debug + Send + Sync;

    // Required method
    fn name() -> &'static str;
}
Expand description

A type-level representation of the kind of a tensor.

Required Associated Types§

type Primitive: Clone + Debug + Send + Sync

The primitive type of the tensor.

Required Methods§

fn name() -> &'static str

The name of the tensor kind.

Object Safety§

This trait is not object safe.

Implementors§

§

impl<B> TensorKind<B> for Bool
where B: Backend,

§

type Primitive = <B as Backend>::BoolTensorPrimitive

§

impl<B> TensorKind<B> for Float
where B: Backend,

§

type Primitive = TensorPrimitive<B>

§

impl<B> TensorKind<B> for Int
where B: Backend,

§

type Primitive = <B as Backend>::IntTensorPrimitive