torch.is_floating_point(input)

torch.is_floating_point(input)

  • input: 输入张量

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

  • 浮点数据类型:torch.float64torch.float32torch.float16torch.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
相关推荐
万邦科技Lafite2 小时前
京东按图搜索京东商品(拍立淘) API (.jd.item_search_img)快速抓取数据
开发语言·前端·数据库·python·电商开放平台·京东开放平台
不会学习的小白O^O2 小时前
双通道深度学习框架可实现从无人机激光雷达点云中提取橡胶树冠
人工智能·深度学习·无人机
丁浩6662 小时前
Python机器学习---6.集成学习与随机森林
python·随机森林·机器学习
charlie1145141913 小时前
现代 Python 学习笔记:Statements & Syntax
笔记·python·学习·教程·基础·现代python·python3.13
Coovally AI模型快速验证5 小时前
OmniNWM:突破自动驾驶世界模型三大瓶颈,全景多模态仿真新标杆(附代码地址)
人工智能·深度学习·机器学习·计算机视觉·自动驾驶·transformer
青春不败 177-3266-05205 小时前
GPT、DeepSeek等大语言模型应用
人工智能·gpt·深度学习·机器学习·语言模型·科研绘图
渡我白衣6 小时前
C++ 同名全局变量:当符号在链接器中“相遇”
开发语言·c++·人工智能·深度学习·microsoft·语言模型·人机交互
麦麦大数据6 小时前
F036 vue+flask中医热性药知识图谱可视化系统vue+flask+echarts+mysql
vue.js·python·mysql·flask·可视化·中医中药
移远通信6 小时前
MQTT协议:物联网时代的通信革命
python·物联网·网络协议
Amo Xiang6 小时前
JavaScript逆向与爬虫实战——基础篇(css反爬之动态字体实现原理及绕过)
爬虫·python·js逆向·动态字体