torch.Tensor.item() 坑,注意只能是一个值,适合返回loss,acc
pytorch---之item()
猜你喜欢
转载自blog.csdn.net/zxyhhjs2017/article/details/90062897
今日推荐
周排行