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