torch.
is_floating_point
Returns True if the data type of input is a floating point data type i.e., one of torch.float64, torch.float32 and torch.float16.
input
torch.float64
torch.float32
torch.float16
input (Tensor) – the input tensor.
此页内容是否对您有帮助
感谢反馈!