Trait burn_core::tensor::TensorKind

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

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

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

Required Associated Types

type Primitive<const D: usize>: Clone + Debug + Sync + 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 Boolwhere B: Backend,

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

impl<B> TensorKind<B> for Floatwhere B: Backend,

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

impl<B> TensorKind<B> for Intwhere B: Backend,

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