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