JitScalarType

class torch.onnx.JitScalarType(value)

Определяемые в torch скалярные типы.

Используйте JitScalarType для преобразования скалярных типов torch и JIT в скалярные типы ONNX.

Примеры

>>> JitScalarType.from_value(torch.ones(1, 2)).onnx_type()
TensorProtoDataType.FLOAT

>>> JitScalarType.from_value(torch_c_value_with_type_float).onnx_type()
TensorProtoDataType.FLOAT

>>> JitScalarType.from_dtype(torch.get_default_dtype).onnx_type()
TensorProtoDataType.FLOAT

dtype() [source]

Преобразование JitScalarType в тип данных torch.

Тип возвращаемого значения

dtype

classmethod from_dtype(dtype) [source]

Преобразование типа данных torch в JitScalarType.

Note: DO NOT USE this API when dtype comes from a torch._C.Value.type() calls.

Ошибка «RuntimeError: INTERNAL ASSERT FAILED at “../aten/src/ATen/core/jit_type_base.h” может возникнуть в нескольких сценариях, где отсутствует информация о форме. Вместо этого используйте API from_value, который безопаснее.

Параметры

dtype (Optional[dtype]) – torch.dtype для создания JitScalarType

Возвращаемое значение

JitScalarType

Исключения

OnnxExporterError – если dtype не является допустимым типом torch.dtype или равен None.

Тип возвращаемого значения

JitScalarType

classmethod from_value(value, default=None) [source]

Создание JitScalarType из скалярного типа значения.

Параметры

• value (Union[None, Value, Tensor]) – Объект для получения скалярного типа.
• default – JitScalarType, который будет возвращен, если скалярный тип не может быть получен из value

Возвращаемое значение

JitScalarType.

Исключения

• OnnxExporterError – если у value нет допустимого скалярного типа и default равен None.
• SymbolicValueError – когда информация value.type() пуста и default равен None

Тип возвращаемого значения

JitScalarType

onnx_compatible() [source]

Возвращает, совместим ли этот JitScalarType с ONNX.

Тип возвращаемого значения

bool

onnx_type() [source]

Преобразование JitScalarType в тип данных ONNX.

Тип возвращаемого значения

TensorProtoDataType

scalar_name() [source]

Преобразование JitScalarType в имя скалярного типа JIT.

Тип возвращаемого значения

Literal[‘Byte’, ‘Char’, ‘Double’, ‘Float’, ‘Half’, ‘Int’, ‘Long’, ‘Short’, ‘Bool’, ‘ComplexHalf’, ‘ComplexFloat’, ‘ComplexDouble’, ‘QInt8’, ‘QUInt8’, ‘QInt32’, ‘BFloat16’, ‘Float8E5M2’, ‘Float8E4M3FN’, ‘Undefined’]

torch_name() [source]

Преобразование JitScalarType в имя типа torch.

Тип возвращаемого значения

Literal[‘bool’, ‘uint8_t’, ‘int8_t’, ‘double’, ‘float’, ‘half’, ‘int’, ‘int64_t’, ‘int16_t’, ‘complex32’, ‘complex64’, ‘complex128’, ‘qint8’, ‘quint8’, ‘qint32’, ‘bfloat16’, ‘float8_e5m2’, ‘float8_e4m3fn’]