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
相关推荐
该用户已不存在8 小时前
Mojo vs Python vs Rust: 2025年搞AI,该学哪个?
后端·python·rust
站大爷IP10 小时前
Java调用Python的5种实用方案:从简单到进阶的全场景解析
python
用户83562907805115 小时前
从手动编辑到代码生成:Python 助你高效创建 Word 文档
后端·python
c8i15 小时前
python中类的基本结构、特殊属性于MRO理解
python
隐语SecretFlow16 小时前
国人自研开源隐私计算框架SecretFlow,深度拆解框架及使用【开发者必看】
深度学习
liwulin050616 小时前
【ESP32-CAM】HELLO WORLD
python
Doris_202316 小时前
Python条件判断语句 if、elif 、else
前端·后端·python
Doris_202316 小时前
Python 模式匹配match case
前端·后端·python
Billy_Zuo17 小时前
人工智能深度学习——卷积神经网络(CNN)
人工智能·深度学习·cnn
这里有鱼汤17 小时前
Python量化实盘踩坑指南:分钟K线没处理好,小心直接亏钱!
后端·python·程序员