快捷方式

torch.isinf

torch.isinf(input) Tensor

测试 input 的每个元素是否为无穷大(正无穷大或负无穷大)。

注意

当复数值的实部或虚部为无穷大时,该复数值为无穷大。

参数

input (Tensor) – 输入张量。

返回值

一个布尔型张量,在 input 为无穷大时为 True,否则为 False

示例

>>> torch.isinf(torch.tensor([1, float('inf'), 2, float('-inf'), float('nan')]))
tensor([False,  True,  False,  True,  False])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获取问题解答

查看资源