jax.numpy.isinf

目录

jax.numpy.isinf#

jax.numpy.isinf(x, /)[源代码][源代码]#

测试元素是否为正无穷或负无穷。

LAX-backend 对 numpy.isinf() 的实现。

原始文档字符串如下。

返回一个与 x 形状相同的布尔数组,其中 x == +/-inf 为 True,否则为 False。

参数:

x (array_like) – 输入值

返回:

y – 当 x 为正无穷或负无穷时为真,否则为假。如果 x 是标量,则这是一个标量。

返回类型:

bool (scalar) or boolean ndarray