torch.is_floating_point(input)

tao3556672024-12-01 1:04

torch.is_floating_point(input)

input: 输入张量

如果输入的数据类型是 浮点数据类型 ,则返回 True。否则返回False。

浮点数据类型：torch.float64、torch.float32、torch.float16 、 torch.bfloat16

python 复制代码

import torch

# 创建一个浮点数张量
float_tensor = torch.tensor([1.0, 2.0, 3.0], dtype=torch.float32)

# 检查张量是否为浮点类型
is_float = torch.is_floating_point(float_tensor)

print(is_float)  # 输出: True

# 创建一个整数张量
int_tensor = torch.tensor([1, 2, 3], dtype=torch.int32)

# 检查张量是否为浮点类型
is_float = torch.is_floating_point(int_tensor)

print(is_float)  # 输出: False

# 创建一个复数张量
complex_tensor = torch.tensor([1 + 2j, 3 - 4j, 5 + 6j], dtype=torch.complex64)

# 检查张量是否为浮点类型
is_float = torch.is_floating_point(complex_tensor)

print(is_float)  # 输出: False