Trait burn_core::tensor::TensorKind

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

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

Required Associated Types

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

Required Methods

fn name() -> &'static str

Implementors

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

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

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

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

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

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