torch.is_floating_point

torch.is_floating_point(input)

Возвращает True, если тип данных input является типом с плавающей точкой, т.е. одним из torch.float64, torch.float32, torch.float16, и torch.bfloat16.

Параметры:

input (Тензор) – входной тензор.