TensorElement

Trait TensorElement

pub trait TensorElement:
    Clone
    + Send
    + Sync
    + 'static
    + Copy {
    const DTYPE: DType;
    const ZERO: Self;
    const ONE: Self;
    const MIN: Self;
    const MAX: Self;
Show 20 methods
    // Required methods
    fn from_f32(val: f32) -> Self;
    fn to_f32(self) -> f32;
    fn from_f64(val: f64) -> Self;
    fn to_f64(self) -> f64;
    fn is_zero(self) -> bool;
    fn infinity() -> Self;
    fn from_u8(val: u8) -> Self;
    fn from_u16(val: u16) -> Self;
    fn from_u32(val: u32) -> Self;
    fn from_u64(val: u64) -> Self;
    fn from_i8(val: i8) -> Self;
    fn from_i16(val: i16) -> Self;
    fn from_i32(val: i32) -> Self;
    fn from_i64(val: i64) -> Self;
    fn from_bool(val: bool) -> Self;
    fn from_f16(val: f16) -> Self;
    fn from_bf16(val: bf16) -> Self;

    // Provided methods
    fn zero() -> Self { ... }
    fn one() -> Self { ... }
    fn dtype() -> DType { ... }
}

Expand description

Trait for types that can be used as tensor elements

Required Associated Constants§

const DTYPE: DType

The data type of this tensor element

const ZERO: Self

The zero/additive identity value for this type

const ONE: Self

The one/multiplicative identity value for this type

const MIN: Self

The minimum value for this type

const MAX: Self

The maximum value for this type

Required Methods§

fn from_f32(val: f32) -> Self

Convert from f32 (for type casting)

fn to_f32(self) -> f32

Convert to f32 (for type casting)

fn from_f64(val: f64) -> Self

Convert from f64 (for type casting)

fn to_f64(self) -> f64

Convert to f64 (for type casting)

fn is_zero(self) -> bool

Check if value is zero

fn infinity() -> Self

Infinity value (for division by zero)

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

Provided Methods§

fn zero() -> Self

Zero value for this type (for backward compatibility)

fn one() -> Self

One value for this type (for backward compatibility)

fn dtype() -> DType

Get the dtype for this type

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementations on Foreign Types§

impl TensorElement for bool

const DTYPE: DType = DType::Bool

const ZERO: Self = false

const ONE: Self = true

const MIN: Self = false

const MAX: Self = true

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for f32

const DTYPE: DType = DType::Fp32

const ZERO: Self = 0f32

const ONE: Self = 1f32

const MIN: Self = -3.40282347E+38f32

const MAX: Self = 3.40282347E+38f32

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for f64

const DTYPE: DType = DType::Fp64

const ZERO: Self = 0f64

const ONE: Self = 1f64

const MIN: Self = -1.7976931348623157E+308f64

const MAX: Self = 1.7976931348623157E+308f64

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for i8

const DTYPE: DType = DType::Int8

const ZERO: Self = 0i8

const ONE: Self = 1i8

const MIN: Self = -128i8

const MAX: Self = 127i8

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for i16

const DTYPE: DType = DType::Int16

const ZERO: Self = 0i16

const ONE: Self = 1i16

const MIN: Self = -32_768i16

const MAX: Self = 32_767i16

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for i32

const DTYPE: DType = DType::Int32

const ZERO: Self = 0i32

const ONE: Self = 1i32

const MIN: Self = -2_147_483_648i32

const MAX: Self = 2_147_483_647i32

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for i64

const DTYPE: DType = DType::Int64

const ZERO: Self = 0i64

const ONE: Self = 1i64

const MIN: Self = -9_223_372_036_854_775_808i64

const MAX: Self = 9_223_372_036_854_775_807i64

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for u8

const DTYPE: DType = DType::Uint8

const ZERO: Self = 0u8

const ONE: Self = 1u8

const MIN: Self = 0u8

const MAX: Self = 255u8

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for u16

const DTYPE: DType = DType::Uint16

const ZERO: Self = 0u16

const ONE: Self = 1u16

const MIN: Self = 0u16

const MAX: Self = 65_535u16

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for u32

const DTYPE: DType = DType::Uint32

const ZERO: Self = 0u32

const ONE: Self = 1u32

const MIN: Self = 0u32

const MAX: Self = 4_294_967_295u32

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for u64

const DTYPE: DType = DType::Uint64

const ZERO: Self = 0u64

const ONE: Self = 1u64

const MIN: Self = 0u64

const MAX: Self = 18_446_744_073_709_551_615u64

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn is_zero(self) -> bool

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for bf16

const DTYPE: DType = DType::Bf16

const ZERO: Self = bf16::ZERO

const ONE: Self = bf16::ONE

const MIN: Self = bf16::MIN

const MAX: Self = bf16::MAX

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn is_zero(self) -> bool

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

impl TensorElement for f16

const DTYPE: DType = DType::Fp16

const ZERO: Self = f16::ZERO

const ONE: Self = f16::ONE

const MIN: Self = f16::MIN

const MAX: Self = f16::MAX

fn from_f32(val: f32) -> Self

fn to_f32(self) -> f32

fn is_zero(self) -> bool

fn from_f64(val: f64) -> Self

fn to_f64(self) -> f64

fn infinity() -> Self

fn from_u8(val: u8) -> Self

fn from_u16(val: u16) -> Self

fn from_u32(val: u32) -> Self

fn from_u64(val: u64) -> Self

fn from_i8(val: i8) -> Self

fn from_i16(val: i16) -> Self

fn from_i32(val: i32) -> Self

fn from_i64(val: i64) -> Self

fn from_bool(val: bool) -> Self

fn from_f16(val: f16) -> Self

fn from_bf16(val: bf16) -> Self

Implementors§