JitScalarType

class torch.onnx.JitScalarType(value)

Типы скаляров, определённые в torch.

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

Примеры::

>>> JitScalarType.from_name("Float").onnx_type()
TensorProtoDataType.FLOAT

dtype() [source]

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

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

dtype

classmethod from_dtype(dtype) [source]

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

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

JitScalarType

classmethod from_name(name) [source]

Преобразовать имя JIT типа скаляра или имя типа torch в ScalarType.

Параметры:

name (Optional[Union[typing_extensions.Literal[Byte, Char, Double, Float, Half, Int, Long, Short, Bool, ComplexHalf, ComplexFloat, ComplexDouble, QInt8, QUInt8, QInt32, BFloat16, Undefined], typing_extensions.Literal[bool, uint8_t, int8_t, double, float, half, int, int64_t, int16_t, complex32, complex64, complex128, qint8, quint8, qint32, bfloat16], str]]) – Имя JIT типа скаляра (Byte) или имя типа torch (uint8_t).

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

ScalarType.

Исключения:

ValueError – если name не является допустимым именем типа скаляра или если оно равно None.

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

JitScalarType

onnx_compatible() [source]

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

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

bool

onnx_type() [source]

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

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

TensorProtoDataType

scalar_name() [source]

Преобразовать ScalarType в имя JIT типа скаляра.

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

typing_extensions.Literal[Byte, Char, Double, Float, Half, Int, Long, Short, Bool, ComplexHalf, ComplexFloat, ComplexDouble, QInt8, QUInt8, QInt32, BFloat16, Undefined]

torch_name() [source]

Преобразовать ScalarType в имя типа torch.

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

typing_extensions.Literal[bool, uint8_t, int8_t, double, float, half, int, int64_t, int16_t, complex32, complex64, complex128, qint8, quint8, qint32, bfloat16]