pytorch---之item()

torch.Tensor.item() 坑,注意只能是一个值,适合返回loss,acc

猜你喜欢

转载自blog.csdn.net/zxyhhjs2017/article/details/90062897