Enum DType 

pub enum DType {
    F64,
    F32,
    F16,
    BF16,
    I64,
    I32,
    I16,
    I8,
    U64,
    U32,
    U16,
    U8,
    Bool,
}

Element type of a tensor.

The types torch.save writes as a plain storage. Complex, quantized and sparse tensors are refused at parse time rather than given a variant here. Deliberately exhaustive: a converter matching on it should fail to compile, not fall through, when a variant is added, so adding one is a breaking change.

Variants

F64

F32

F16

BF16

I64

I32

I16

I8

U64

U32

U16

U8

Bool

One byte per element, holding 0 or 1.

Implementations

impl DType

pub const fn size(self) -> usize

Size of one element in bytes.

Trait Implementations

impl Clone for DType

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for DType

impl Debug for DType

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

Formats the value using the given formatter.

impl Eq for DType

impl Hash for DType

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

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: &Self) -> bool

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for DType