DataType

class onnx_ir.DataType(value)

ONNX 张量数据类型的枚举，定义在 onnx.TensorProto 中。

classmethod from_numpy(dtype)

返回 numpy dtype 对应的 ONNX 数据类型。

引发:

TypeError – 如果数据类型不受 ONNX 支持。

参数:

dtype (dtype)

返回类型:

DataType

classmethod from_short_name(short_name)

返回短名称对应的 ONNX 数据类型。

引发:

TypeError – 如果数据类型没有可用的短名称。

参数:

short_name (str)

返回类型:

DataType

property itemsize: float

返回数据类型的大小（以字节为单位）。

property bitwidth: int

返回数据类型的位宽。

0.1.2 版本新增。

引发:

TypeError – 如果数据类型不受支持。

property exponent_bitwidth: int

返回浮点类型的指数位宽。

在 0.1.8 版本中添加。

引发:

TypeError – 如果数据类型不受支持。

property mantissa_bitwidth: int

返回浮点类型的尾数位宽。

在 0.1.8 版本中添加。

引发:

TypeError – 如果数据类型不受支持。

property eps: int | floating[Any]

对于 ONNX 数据类型，返回 1.0 与下一个比 1.0 稍大的可表示浮点数之间的差值。

对于整数，返回 1。

在 0.1.8 版本中添加。

引发:

TypeError – 如果数据类型不是数值数据类型。

property tiny: int | floating[Any]

返回 ONNX 数据类型支持的最小正非零值。

对于整数，返回 1。

在 0.1.8 版本中添加。

引发:

TypeError – 如果数据类型不是数值数据类型。

property min: int | floating[Any]

返回 ONNX 数据类型支持的最小可表示值。

在 0.1.8 版本中添加。

引发:

TypeError – 如果数据类型不是数值数据类型。

property max: int | floating[Any]

返回 ONNX 数据类型支持的最大可表示值。

在 0.1.8 版本中添加。

引发:

TypeError – 如果数据类型不是数值数据类型。

property precision: int

如果支持，返回 ONNX dtype 的精度。

对于浮点数，返回这种浮点数精确到的小数位数。对于整数，返回 0。

在 0.1.8 版本中添加。

引发:

TypeError – 如果数据类型不是数值数据类型。

property resolution: int | floating[Any]

如果支持，返回 ONNX dtype 的分辨率。

返回此类型的近似十进制分辨率，即 10**-precision。对于整数，返回 1。

10**-precision。对于整数，返回 1。

在 0.1.8 版本中添加。

引发:

TypeError – 如果数据类型不是数值数据类型。

numpy()

返回 ONNX 数据类型对应的 numpy dtype。

引发:

TypeError – 如果数据类型不受 numpy 支持。

返回类型:

dtype

short_name()

返回数据类型的短名称。

短名称是一个字符串，用于以更紧凑的形式表示数据类型。例如，DataType.FLOAT 的短名称是“f32”。要获取相应的原始数据类型，请对字符串调用 from_short_name。

命名参考：https://github.com/pytorch/pytorch/blob/4bead7b85ea4160243c74109e0ce9bb80686d016/torch/utils/_dtype_abbrs.py

引发:

TypeError – 如果数据类型没有可用的短名称。

返回类型:

str

is_floating_point()

如果数据类型是浮点类型，则返回 True。

返回类型:

布尔

is_integer()

如果数据类型是整数，则返回 True。

在 0.1.4 版本中添加。

返回类型:

布尔

is_signed()

如果数据类型是有符号类型，则返回 True。

在 0.1.4 版本中添加。

返回类型:

布尔

is_string()

如果数据类型是字符串类型，则返回 True。

在 0.1.8 版本中添加。

返回类型:

布尔

UNDEFINED = 0

FLOAT = 1

UINT8 = 2

INT8 = 3

UINT16 = 4

INT16 = 5

INT32 = 6

INT64 = 7

STRING = 8

BOOL = 9

FLOAT16 = 10

DOUBLE = 11

UINT32 = 12

UINT64 = 13

COMPLEX64 = 14

COMPLEX128 = 15

BFLOAT16 = 16

FLOAT8E4M3FN = 17

FLOAT8E4M3FNUZ = 18

FLOAT8E5M2 = 19

FLOAT8E5M2FNUZ = 20

UINT4 = 21

INT4 = 22

FLOAT4E2M1 = 23

FLOAT8E8M0 = 24