torch.is_floating_point(input)
torch.is_floating_point(input)
- input: 输入张量
如果输入的数据类型是 浮点数据类型 ,则返回 True。否则返回False。
- 浮点数据类型：
torch.float64、torch.float32、torch.float16、torch.bfloat16
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
原文地址：https://blog.csdn.net/m0_63070489/article/details/144041081
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！