is_floating_point

paddle. is_floating_point ( x ) [源代码]

判断输入 Tensor 的数据类型是否为浮点类型。

参数

  • x (Tensor) - 输入的 Tensor。

返回

bool,输入 tensor 的数据类型为浮点数类型则为 True,反之为 False.

代码示例

import paddle

x = paddle.arange(1., 5., dtype='float32')
y = paddle.arange(1, 5, dtype='int32')
print(paddle.is_floating_point(x))
# True
print(paddle.is_floating_point(y))
# False