torch.promote_types

torch.promote_types(type1, type2) → dtype

Возвращает torch.dtype с наименьшим размером и скалярным типом, который не меньше и не ниже ни type1 ни type2. Дополнительную информацию о логике повышения типа см. в документации.

Параметры:

• type1 (torch.dtype) –
• type2 (torch.dtype) –

Пример:

>>> torch.promote_types(torch.int32, torch.float32)
torch.float32
>>> torch.promote_types(torch.uint8, torch.long)
torch.long