API

 torch / torch


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 of torch.float64, torch.float32 and torch.float16.

Parameters

input (Tensor) – the input tensor.


此页内容是否对您有帮助