因为keepdim维度和输入相同,所以并没有自动降维仍旧是2维,但是target是1维(一维数字嘛),所以需要target维度变成和pred一样的维度。view_as 返回被视作与给定的tensor相同大小的原tensor。
由此看完全代码完全等效于图片中的,亲测有效。max自动降维,就不需要target升维了。
因为keepdim维度和输入相同,所以并没有自动降维仍旧是2维,但是target是1维(一维数字嘛),所以需要target维度变成和pred一样的维度。view_as 返回被视作与给定的tensor相同大小的原tensor。
由此看完全代码完全等效于图片中的,亲测有效。max自动降维,就不需要target升维了。