Trait burn::tensor::TensorKind

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

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

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

Required Associated Types§

type Primitive<const D: usize>: Clone + Debug + Send

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<const D: usize> = <B as Backend>::BoolTensorPrimitive<D>

§

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

§

type Primitive<const D: usize> = TensorPrimitive<B, D>

§

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

§

type Primitive<const D: usize> = <B as Backend>::IntTensorPrimitive<D>