欢迎您访问365答案网,请分享给你的朋友!
生活常识 学习资料

Pytorchtensor避免nan解决办法

时间:2023-05-15

使用pytorch处理数据运算是,如矩阵涉及乘除法,如矩阵中存在nan数据,则结果中存在nan,解决方案是使用torch.where()将nan值替换为0或其他值:

torch.where(torch.isnan(a), torch.full_like(a, 0), a)

Copyright © 2016-2020 www.365daan.com All Rights Reserved. 365答案网 版权所有 备案号:

部分内容来自互联网,版权归原作者所有,如有冒犯请联系我们,我们将在三个工作时内妥善处理。