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 >>> import mindspore.ops as 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