Enum DType 

pub enum DType {
    F16,
    BF16,
    F32,
    F64,
    U8,
    U32,
    I64,
}

Re-export core types. Enum of all supported element data types.

This is stored inside every tensor so we can dispatch operations to the correct typed implementation at runtime.

Variants

F16

BF16

F32

F64

U8

U32

I64

Implementations

impl DType

pub fn size_in_bytes(&self) -> usize

Size of one element in bytes.

pub fn is_float(&self) -> bool

Whether this dtype is a floating-point type (needed for gradient tracking).

pub fn is_half(&self) -> bool

Whether this is a half-precision type (F16 or BF16).

Trait Implementations

impl Clone for DType

fn clone(&self) -> DType

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DType

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Display for DType

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Hash for DType

fn hash<__H>(&self, state: &mut __H)

where __H: Hasher,

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for DType

fn eq(&self, other: &DType) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for DType

impl Eq for DType

impl StructuralPartialEq for DType