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