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])