请选择 进入手机版 | 继续访问电脑版
查看: 202|回复: 2

Pytorch操作检测NaNs

[复制链接]

9

主题

47

帖子

123

积分

注册会员

Rank: 2

积分
123
发表于 2018-10-15 16:59:11 | 显示全部楼层 |阅读模式
是否有一个Pytorch-internal过程来检测张量中的NaNs ? Tensorflow 有 tf.is_nan 和tf.check_numerics操作
Pytorch也有类似的吗?但是好像我没有找到。我特别寻找Pytorch的内部例程,因为我希望这种情况发生在GPU和CPU上。
这不包括numpy(如:
  1. np.isnan(sometensor.numpy()).any()
复制代码



回复

使用道具 举报

44

主题

139

帖子

381

积分

论坛管理

Rank: 4

积分
381
发表于 2018-10-15 17:02:46 | 显示全部楼层
可以利用这个
  1. >>> x = torch.tensor([1, 2, np.nan])
  2. tensor([  1.,   2., nan.])
  3. >>> x != x
  4. tensor([ 0,  0,  1], dtype=torch.uint8)
复制代码


在pytorch 0.4有:torch.isnan
  1. >>> torch.isnan(x)
  2. tensor([ 0,  0,  1], dtype=torch.uint8)
复制代码
回复

使用道具 举报

6

主题

21

帖子

65

积分

注册会员

Rank: 2

积分
65
发表于 2018-10-16 08:47:12 | 显示全部楼层
了解了,我们可以多看看别人的错误,让自己少犯错
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表