Python PyTorch promote_types用法及代碼示例

本文簡要介紹python語言中 torch.promote_types 的用法。

用法:

torch.promote_types(type1, type2) → dtype

參數：

• type1(torch.dtype) -

• type2(torch.dtype) -

返回具有最小大小和標量類型的 torch.dtype ，不小於或低於 type1 或 type2 。有關類型提升邏輯的更多信息，請參閱類型提升文檔。

例子：

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