pytorch处理inf和nan数值

刺骨的言语ヽ痛彻心扉 2022-12-04 07:46 385阅读 0赞

在构建网络框架后，运行代码，发现很多tensor出现了inf值或者nan,在很多博客上没有找到对应的解决方法，大部分是基于numpy写的，比较麻烦。下面基于torch BIF函数实现替换这2个值。

a = torch.Tensor([[1, 2, np.nan], [np.inf, np.nan, 4], [3, 4, 5]])
    
    a
    Out[158]: 
    tensor([[1., 2., nan],
            [inf, nan, 4.],
            [3., 4., 5.]])

下面把nan值还为0：

a = torch.where(torch.isnan(a), torch.full_like(a, 0), a)
    
    a
    Out[160]: 
    tensor([[1., 2., 0.],
            [inf, 0., 4.],
            [3., 4., 5.]])

接着把inf替换为1：

a = torch.where(torch.isinf(a), torch.full_like(a, 0), a)
    
    a
    Out[162]: 
    tensor([[1., 2., 0.],
            [0., 0., 4.],
            [3., 4., 5.]])

# 简单回顾 #

**tips**:对于某些tensor,可能已经开启了grad功能，需要把它先转为普通tensor（使用.data）

torch.where(condition,T,F) 函数有三个输入值，

第一个是判断条件，

第二个是符合条件的设置值，

第三个是不符合条件的设置值

torch.full_like(input, fill_value, …) 返回与input相同size，单位值为fill_value的矩阵
    
    #如下面这个例子，a为3*3的tensor
    b =torch.full_like(a, 0,)
    
    b
    Out[165]: 
    tensor([[0., 0., 0.],
            [0., 0., 0.],
            [0., 0., 0.]])

发表评论取消回复

表情：

评论列表（有 0 条评论，385人围观）

还没有评论，来说两句吧...

相关阅读

相关 pandas query如何处理NaN

目录 01现象 02原因 03解决方案思路一思路二 -------------------- 01现象 dataframe.query("col1 !=

刺骨的言语ヽ痛彻心扉/ 2023年01月17日 07:41/ 0 赞/ 103 阅读

相关 matlab中数据变为nan,字符转化为数值型中出现NAN

我将字符型转化为数值型，然后画图，结果图形没有曲线，这是怎么回事？我用的函数是str2double和str2num都试了，都不行。我的程序如下,其中的E2(i)的值我用matl

忘是亡心i/ 2023年01月15日 03:41/ 0 赞/ 169 阅读

相关 pytorch处理inf和nan数值

在构建网络框架后，运行代码，发现很多tensor出现了inf值或者nan,在很多博客上没有找到对应的解决方法，大部分是基于numpy写的，比较麻烦。下面基于torch B

刺骨的言语ヽ痛彻心扉/ 2022年12月04日 07:46/ 0 赞/ 386 阅读

相关 javaScript中NaN不和任何数值相等，包括它本身

[JavaScript中的NaN、Infinity、null和undefined][JavaScript_NaN_Infinity_null_undefined] 直接上代码

偏执的太偏执、/ 2022年11月05日 13:50/ 0 赞/ 75 阅读

相关 Infinity和NaN

今天，偶然发现当double类型的数除以0的时候编译器竟然没有报错，运行结果为Infinity。 public static void main(String[] a

蔚落/ 2022年08月02日 12:56/ 0 赞/ 140 阅读

相关 NaN,非数值

NaN,即非数值，是一个特殊的数值，这个数值适用于表示一个本来要返回的操作数未返回数值的情况（这样子就不会抛出错误了）。例如，在其他的编程语言中，任何数值除以0都是会导致错误的

怼烎@/ 2022年06月16日 11:08/ 0 赞/ 201 阅读

相关 c/c++中的inf与nan

inf ：infinity （linux）等同于 \INF:infinity (windows) nan ：not a number 等同于 \IND:indetermi

╰+哭是因爲堅強的太久メ/ 2022年06月01日 07:18/ 0 赞/ 145 阅读

相关 Python numpy中的nan和inf，nan替换成对应的均值

nan：not a number inf：infinity;正无穷 numpy中的nan和inf都是float类型 ![watermark_type_ZmFuZ3p

喜欢ヅ旅行/ 2022年03月18日 03:40/ 0 赞/ 171 阅读

相关 Tenflow bug（二）——InvalidArgumentError: LossTensor is inf or nan : Tensor had NaN values

之前在TensorFlow中实现不同的神经网络，作为新手，发现经常会出现计算的loss中，出现Nan值的情况，总的来说，TensorFlow中出现Nan值的情况有两

谁借莪１个温暖的怀抱￠/ 2022年01月13日 07:13/ 0 赞/ 278 阅读

相关 Python 数据清洗--处理Nan

参考：[http://blog.sina.com.cn/s/blog\_13050351e0102xfis.html][http_blog.sina.com.cn_s_blog

- 日理万妓/ 2022年01月12日 01:37/ 0 赞/ 269 阅读