Trait burn_core::tensor::ElementConversion

pub trait ElementConversion {
    // Required methods
    fn from_elem<E>(elem: E) -> Self
       where E: ToPrimitive;
    fn elem<E>(self) -> E
       where E: Element;
}

Element conversion trait for tensor.

Required Methods

fn from_elem<E>(elem: E) -> Selfwhere E: ToPrimitive,

Converts an element to another element.

Arguments

• elem - The element to convert.

Returns

The converted element.

fn elem<E>(self) -> Ewhere E: Element,

Converts and returns the converted element.

Object Safety

This trait is not object safe.

Implementations on Foreign Types

impl ElementConversion for f32

fn from_elem<E>(elem: E) -> f32where E: ToPrimitive,

fn elem<E>(self) -> Ewhere E: Element,

impl ElementConversion for f64

fn from_elem<E>(elem: E) -> f64where E: ToPrimitive,

fn elem<E>(self) -> Ewhere E: Element,

impl ElementConversion for i8

fn from_elem<E>(elem: E) -> i8where E: ToPrimitive,

fn elem<E>(self) -> Ewhere E: Element,

impl ElementConversion for i16

fn from_elem<E>(elem: E) -> i16where E: ToPrimitive,

fn elem<E>(self) -> Ewhere E: Element,

impl ElementConversion for i32

fn from_elem<E>(elem: E) -> i32where E: ToPrimitive,

fn elem<E>(self) -> Ewhere E: Element,

impl ElementConversion for i64

fn from_elem<E>(elem: E) -> i64where E: ToPrimitive,

fn elem<E>(self) -> Ewhere E: Element,

impl ElementConversion for u8

fn from_elem<E>(elem: E) -> u8where E: ToPrimitive,

fn elem<E>(self) -> Ewhere E: Element,

impl ElementConversion for u32

fn from_elem<E>(elem: E) -> u32where E: ToPrimitive,

fn elem<E>(self) -> Ewhere E: Element,

Implementors

impl ElementConversion for bf16

impl ElementConversion for f16