Trait burn_tensor::TensorKind

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

Required Associated Types

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

Implementors

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

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

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

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

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

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