假设A为tensor数组
在网络训练过程中，求tensor数组的平均值时，可以使用torch.nanmean(A)来剔除数组中的异常数值或无穷大数值。这个函数等价于torch.mean(A[~A.isnan()])
剔除tensor数组中的异常数值
猜你喜欢
转载自blog.csdn.net/qq_41598072/article/details/130346047
今日推荐
周排行