torch.promote_types

torch.promote_types(type1, type2) → dtype

返回一个类型（dtype），它的大小和标量种类不小于 type1 或 type2，并且具有最小的 size 和 scalar kind。关于类型提升逻辑的更多信息，请参阅类型提升 文档。

参数

• type1 (torch.dtype) –

• type2 (torch.dtype) –

示例

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