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