Python PyTorch result_type用法及代碼示例

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

用法:

torch.result_type(tensor1, tensor2) → dtype

參數：

• tensor1(Tensor或者數字) -輸入張量或數字

• tensor2(Tensor或者數字) -輸入張量或數字

返回對提供的輸入張量執行算術運算所產生的 torch.dtype 。有關類型提升邏輯的更多信息，請參閱類型提升文檔。

例子：

>>> 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