Trait tch::kind::Element

pub unsafe trait Element: Clone {
    const KIND: Kind;
    const ZERO: Self;
}

Kinds for tensor elements

Safety

The specified Kind must be for a type that has the same length as Self.

Required Associated Constants

const KIND: Kind

const ZERO: Self

Implementations on Foreign Types

impl Element for f32

const KIND: Kind = Kind::Float

const ZERO: Self = 0f32

impl Element for f64

const KIND: Kind = Kind::Double

const ZERO: Self = 0f64

impl Element for bf16

const KIND: Kind = Kind::Half

const ZERO: Self = half::bf16::ZERO

impl Element for i64

const KIND: Kind = Kind::Int64

const ZERO: Self = 0i64

impl Element for bool

const KIND: Kind = Kind::Bool

const ZERO: Self = false

impl Element for i32

const KIND: Kind = Kind::Int

const ZERO: Self = 0i32

impl Element for u8

const KIND: Kind = Kind::Uint8

const ZERO: Self = 0u8

impl Element for f16

const KIND: Kind = Kind::Half

const ZERO: Self = half::f16::ZERO

impl Element for i8

const KIND: Kind = Kind::Int8

const ZERO: Self = 0i8

impl Element for i16

const KIND: Kind = Kind::Int16

const ZERO: Self = 0i16

Implementors