Enum DType 

pub enum DType {
    Float32,
    Float16,
    Float64,
    Int8,
    Int16,
    Int32,
    Int64,
    Uint8,
    Uint16,
    Uint32,
    Uint64,
    Bool,
}

Numerical data type for array contents.

Corresponds to jax-js DType enum. Supports basic types that can be efficiently worked with on the web.

Variants

Float32

32-bit floating point

Float16

16-bit floating point (stored as u16 bits)

Float64

64-bit floating point

Int8

8-bit signed integer

Int16

16-bit signed integer

Int32

32-bit signed integer

Int64

64-bit signed integer

Uint8

8-bit unsigned integer

Uint16

16-bit unsigned integer

Uint32

32-bit unsigned integer

Uint64

64-bit unsigned integer

Bool

Boolean (stored as 1-byte value)

Implementations

impl DType

pub const fn byte_width(self) -> usize

Returns the byte width of this dtype.

pub const fn is_float(self) -> bool

Returns true if this is a floating-point dtype.

pub const fn is_int(self) -> bool

Returns true if this is an integer dtype.

pub const fn is_signed(self) -> bool

Returns true if this is a signed integer dtype.

pub const fn is_unsigned(self) -> bool

Returns true if this is an unsigned integer dtype.

pub fn promote(dtype1: DType, dtype2: DType) -> DType

Promotes two dtypes according to JAX’s type promotion rules.

Type lattice: bool -> uint8 -> uint16 -> uint32 -> int8 -> int16 -> int32 -> float16 -> float32 -> float64

Examples

assert_eq!(DType::promote(DType::Bool, DType::Int32), DType::Int32);
assert_eq!(DType::promote(DType::Uint32, DType::Int32), DType::Int32);
assert_eq!(DType::promote(DType::Int32, DType::Float16), DType::Float16);
assert_eq!(DType::promote(DType::Float16, DType::Float32), DType::Float32);

pub fn cast_from_f32(self, value: f32) -> f32

Cast a float32 value to this dtype (returns as f32 for storage).

impl DType

pub fn from_str(s: &str) -> Option<DType>

Parse a string into a DType.

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

Formats the value using the given formatter.

impl Display for DType

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

Formats the value using the given formatter.

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 Copy for DType

impl Eq for DType

impl StructuralPartialEq for DType