DataType类
数据类型的枚举类，提供了GE的data type定义，方便实现converter函数时调用。DataType类的具体定义如下：
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
class DataType: DT_FLOAT = 0 # float type DT_FLOAT16 = 1 # fp16 type DT_INT8 = 2 # int8 type DT_INT16 = 6 # int16 type DT_UINT16 = 7 # uint16 type DT_UINT8 = 4 # uint8 type DT_INT32 = 3 # int32 type DT_INT64 = 9 # int64 type DT_UINT32 = 8 # unsigned int32 DT_UINT64 = 10 # unsigned int64 DT_BOOL = 12 # bool type DT_DOUBLE = 11 # double type DT_STRING = 13 # string type DT_DUAL_SUB_INT8 = 14 # dual output int8 type DT_DUAL_SUB_UINT8 = 15 # dual output uint8 type DT_COMPLEX64 = 16 # complex64 type DT_COMPLEX128 = 17 # complex128 type DT_QINT8 = 18 # qint8 type DT_QINT16 = 19 # qint16 type DT_QINT32 = 20 # qint32 type DT_QUINT8 = 21 # quint8 type DT_QUINT16 = 22 # quint16 type DT_RESOURCE = 23 # resource type DT_STRING_REF = 24 # string ref type DT_DUAL = 25 # dual output type DT_VARIANT = 26 # dt_variant type DT_BF16 = 27 # bf16 type DT_UNDEFINED = 28 # Used to indicate a DataType field has not been set. DT_INT4 = 29 # int4 type DT_UINT1 = 30 # uint1 type DT_INT2 = 31 # int2 type DT_UINT2 = 32 # uint2 type DT_COMPLEX32 = 33 # complex64 type DT_MAX = 34 # Mark the boundaries of data types |
调用示例如下：
1 2 |
from torchair.ge import DataType Cast(Tensor, dst_type=DataType.DT_BF16) |
父主题： GE相关类和接口
