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