oneflow.is_floating_point¶
-
oneflow.
is_floating_point
(input)¶ Returns True if the data type of input is a floating point data type i.e., one of oneflow.float64 , oneflow.float32 , oneflow.float16, and oneflow.bfloat16.
- Parameters
input (Tensor) – the input tensor.
For example:
>>> import oneflow as flow >>> input = flow.tensor([1, 2, 3, 4, 5], dtype=flow.int) >>> output = flow.is_floating_point(input) >>> output False