torch.promote_types¶
- torch.promote_types(type1, type2) dtype ¶
返回大小最小且标量类型不小于 type1 或 type2 的
torch.dtype
。有关类型提升逻辑的更多信息,请参阅类型提升 文档。- 参数
type1 (
torch.dtype
) –type2 (
torch.dtype
) –
示例
>>> torch.promote_types(torch.int32, torch.float32) torch.float32 >>> torch.promote_types(torch.uint8, torch.long) torch.long