pytorch中torch.isnan()和torch.isfinite()怎么用

发布时间:2021-07-19 10:04:06 作者:chen
来源:亿速云 阅读:397

这篇文章主要讲解了“pytorch中torch.isnan()和torch.isfinite()怎么用”,文中的讲解内容简单清晰,易于学习与理解,下面请大家跟着小编的思路慢慢深入,一起来研究和学习“pytorch中torch.isnan()和torch.isfinite()怎么用”吧!

1.torch.isfinite()

import torch
num = torch.tensor(1)  # 数字1
res = torch.isfinite(num)
print(res)
'''
输出:
tensor(True)
'''

这个num必须是tensor

import torch
num = torch.tensor(float('inf')) # 正无穷大
res = torch.isfinite(num)
print(res)
'''
输出:
tensor(False)
'''
import torch
num = torch.tensor(float('-inf')) # 负无穷大
res = torch.isfinite(num)
print(res)
'''
输出:
tensor(False)
'''
import torch
num = torch.tensor(float('nan')) # 空
res = torch.isfinite(num)
print(res)
'''
输出:
tensor(False)
'''

2.torch.isnan()

import torch
res=torch.isnan(torch.tensor([1, float('inf'), 2, float('-inf'), float('nan')]))
print(res)
'''
输出:
tensor([False, False, False, False,  True])
'''

可以看出torch.isnan()是用来判断输入的张量是否为空的函数,当输入为空是,返回True。

感谢各位的阅读,以上就是“pytorch中torch.isnan()和torch.isfinite()怎么用”的内容了,经过本文的学习后,相信大家对pytorch中torch.isnan()和torch.isfinite()怎么用这一问题有了更深刻的体会,具体使用情况还需要大家实践验证。这里是亿速云,小编将为大家推送更多相关知识点的文章,欢迎关注!

推荐阅读:
  1. pytorch模型 .pt, .pth, .pkl有什么区别及如何保存
  2. 如何实现Pytorch通过保存为ONNX模型转TensorRT5

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

pytorch

上一篇:SVG怎么实现复杂线条动画

下一篇:python中PaddleOCR库的用法

相关阅读

您好,登录后才能下订单哦!

密码登录
登录注册
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》