使用pytorch处理数据运算是,如矩阵涉及乘除法,如矩阵中存在nan数据,则结果中存在nan,解决方案是使用torch.where()将nan值替换为0或其他值:
torch.where(torch.isnan(a), torch.full_like(a, 0), a)
使用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答案网 版权所有 备案号:
部分内容来自互联网,版权归原作者所有,如有冒犯请联系我们,我们将在三个工作时内妥善处理。