torch.is_floating_point
-
torch.
is_floating_point
(input) -> (bool) Returns True if the data type of
input
is a floating point data type i.e., one oftorch.float64
,torch.float32
andtorch.float16
.- Parameters
input (Tensor) – the PyTorch tensor to test