Trait burn_core::tensor::TensorKind

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

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

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

Required Associated Types§

source

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

The primitive type of the tensor.

Required Methods§

source

fn name() -> &'static str

The name of the tensor kind.

Object Safety§

This trait is not object safe.

Implementors§

source§

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

§

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

source§

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

§

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

source§

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

§

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