Loading

pytorch 判断tensor是否相等

torch.equal();该方式判断tensor,形状和值是否完全相等。不太适用于浮点
torch.allclose();该方式判断tensor,形状和值是否足够接近,适用于浮点对比
torch.set_printoptions(precision=10) 打印设置;

posted @ 2022-04-10 20:53  lif323  阅读(339)  评论(0)    收藏  举报