Trait burn_tch::TchElement

pub trait TchElement: Element + Element { }

The element type for the tch backend.

Object Safety

This trait is not object safe.

Implementations on Foreign Types

impl TchElement for f32

impl TchElement for f64

impl TchElement for i16

impl TchElement for i32

impl TchElement for i64

impl TchElement for u8

impl TchElement for bf16

impl TchElement for f16

Implementors