JitScalarType

class torch.onnx.JitScalarType(value)

Scalar types defined in torch.

Use JitScalarType to convert from torch and JIT scalar types to ONNX scalar types.

Examples::

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

dtype()

Convert a ScalarType to a torch dtype.

classmethod from_dtype(dtype)

Convert a torch dtype to ScalarType.

classmethod from_name(name)

Convert a JIT scalar type or torch type name to ScalarType.

Parameters:

name – JIT scalar type name (Byte) or torch type name (uint8_t).

Returns:

ScalarType.

Raises:

ValueError – if name is not a valid scalar type name or if it is None.

onnx_compatible()

Return whether this ScalarType is compatible with ONNX.

onnx_type()

Convert a ScalarType to an ONNX data type.

scalar_name()

Convert a ScalarType to a JIT scalar type name.

torch_name()

Convert a ScalarType to a torch type name.