Enum DType

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

The different types of elements allowed in tensors.

Variants

U8

U32

I64

BF16

F16

F32

F64

Implementations

impl DType

pub fn as_str(&self) -> &'static str

String representation for dtypes.

pub fn size_in_bytes(&self) -> usize

The size used by each element in bytes, i.e. 1 for U8, 4 for F32.

pub fn is_int(&self) -> bool

pub fn is_float(&self) -> bool

Trait Implementations

impl Clone for DType

fn clone(&self) -> DType

Returns a copy 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

Formats the value using the given formatter.

impl From<DType> for Dtype

fn from(value: DType) -> Self

Converts to this type from the input type.

impl FromStr for DType

type Err = DTypeParseError

The associated error which can be returned from parsing.

fn from_str(s: &str) -> Result<Self, Self::Err>

Parses a string s to return a value of this type.

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: &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 TryFrom<Dtype> for DType

type Error = Error

The type returned in the event of a conversion error.

fn try_from(value: Dtype) -> Result<Self>

Performs the conversion.

impl Copy for DType

impl Eq for DType

impl StructuralPartialEq for DType