Trait burn_tensor::TensorKind

pub trait TensorKind<B: Backend>: Clone + Debug {
    type Primitive<const D: usize>: Clone + Debug + 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 + 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: 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>::FloatTensorPrimitive<D>

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

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