returnn.frontend.dtype

DType helpers

returnn.frontend.dtype.get_default_float_dtype() → str

https://data-apis.org/array-api/latest/API_specification/data_types.html#default-data-types

Returns:

default dtype for float

returnn.frontend.dtype.get_default_int_dtype() → str

https://data-apis.org/array-api/latest/API_specification/data_types.html#default-data-types

Returns:

default dtype for int

returnn.frontend.dtype.get_default_array_index_dtype() → str

https://data-apis.org/array-api/latest/API_specification/data_types.html#default-data-types

Returns:

default dtype for array index - currently just the same as get_default_int_dtype()

returnn.frontend.dtype.is_float_dtype(dtype: str) → bool

Returns:

whether the dtype is float, e.g. it supports backprop etc