mindspore.ops.is_floating_point

查看源文件
mindspore.ops.is_floating_point(input)[源代码]

判断 input 的dtype是否是浮点数据类型,包括mindspore.float64,mindspore.float32,mindspore.float16。

参数:
  • input (Tensor) - 输入Tensor。

返回:

Bool,如果 input 的dtype是浮点数据类型,则返回 True ,否则返回 False

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore as ms
>>> from mindspore import ops
>>> from mindspore import Tensor
>>> x = ms.Tensor([1, 2, 3], ms.float32)
>>> y = ms.Tensor([1, 2, 3], ms.int64)
>>> output = ops.is_floating_point(x)
>>> output2 = ops.is_floating_point(y)
>>> print(output)
True
>>> print(output2)
False