使用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)