isinf

paddle.isinf ( x, name=None ) [源代码]

返回输入tensor的每一个值是否为 +/-INF 。

参数

• x (Tensor): 输入的 Tensor ，数据类型为：float16、float32、float64、int32、int64。

• name (str, 可选): 操作的名称(可选，默认值为None）。更多信息请参见 Name 。

返回

Tensor, 每个元素是一个bool值，表示输入 x 的每个元素是否为 +/-INF 。

代码示例

import paddle
import numpy as np
x_np = np.array([float('-inf'), -2, 3.6, float('inf'), 0, float('-nan'), float('nan')])
x = paddle.to_tensor(x_np)
out = paddle.isinf(x)
print(out)  # [True, False, False, True, False, False, False]