isfinite

paddle.fluid.layers.isfinite(x)[源代码]

注意:此算子的输入 Tensor / LoDTensor 数据类型必须为 int32 / float / double 之一。

测试 x 是否包含无穷值(即 nan 或 inf)。若元素均为有穷数,返回真;否则返回假。

参数:
  • x(variable) : 变量,包含被测试的 Tensor / LoDTensor。
返回:
  • Variable (Tensor / LoDTensor),此 Tensor 变量包含一个 bool 型结果。
返回类型
  • Variable (Tensor / LoDTensor),一个包含 Tensor 的变量。

代码示例

import paddle.fluid as fluid
import numpy

# Graph Organizing
var = fluid.layers.data(name="data", shape=(4, 6), dtype="float32")
output = fluid.layers.isfinite(var)

# Create an executor using CPU as an example
exe = fluid.Executor(fluid.CPUPlace())
exe.run(fluid.default_startup_program())

# Execute
img = numpy.array((4, 6)).astype(numpy.float32)
res, = exe.run(fluid.default_main_program(), feed={'data':img}, fetch_list=[output])
print(res)  # Output Value: [ True]