torch.result_type

torch.result_type(tensor1, tensor2) → dtype

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

Параметры:

• tensor1 (Tensor или Number) – входной тензор или число
• tensor2 (Tensor или Number) – входной тензор или число

Пример:

>>> torch.result_type(torch.tensor([1, 2], dtype=torch.int), 1.0)
torch.float32
>>> torch.result_type(torch.tensor([1, 2], dtype=torch.uint8), torch.tensor(1))
torch.uint8