torch.is_floating_point

torch.is_floating_point(input)

如果 input 的資料類型是浮點數資料類型 (即 torch.float64、torch.float32、torch.float16 和 torch.bfloat16 其中之一)，則傳回 True。

參數

input (Tensor) – 輸入張量。