torch中eq用法-view_as

因为keepdim维度和输入相同,所以并没有自动降维仍旧是2维,但是target是1维(一维数字嘛),所以需要target维度变成和pred一样的维度。view_as 返回被视作与给定的tensor相同大小的原tensor。 

由此看完全代码完全等效于图片中的,亲测有效。max自动降维,就不需要target升维了。

猜你喜欢

转载自blog.csdn.net/weixin_42133481/article/details/114291040