Rate this Page
★★★★★
torch.is_floating_point#
- torch.is_floating_point(input:Tensor)→bool#
Returns True if the data type of
inputis a floating point data type i.e.,one oftorch.float64,torch.float32,torch.float16, andtorch.bfloat16.- Parameters
input (Tensor) – the input tensor.
Example:
>>>torch.is_floating_point(torch.tensor([1.0,2.0,3.0]))True>>>torch.is_floating_point(torch.tensor([1,2,3],dtype=torch.int32))False>>>torch.is_floating_point(torch.tensor([1.0,2.0,3.0],dtype=torch.float16))True>>>torch.is_floating_point(torch.tensor([1,2,3],dtype=torch.complex64))False
On this page
