prim = ops.IsInf() out = prim(input)
等价于
ops.isinf(input)
更多详情请查看: mindspore.ops.isinf() 。
mindspore.ops.isinf()