PyTorch 张量 reshape: x = x.view(x.size(0), -1) 解释
这行代码的意思是将张量 x 的形状改变为 (x.size(0), -1) 的形状。其中 x.size(0) 表示张量 x 的第一个维度的大小,-1 表示根据张量的总元素个数自动计算出第二个维度的大小。
这行代码常用于将一个多维张量展平为一个一维张量。例如,如果 x 的形状为 (2, 3, 4),那么经过这行代码后,x 的形状将变为 (2, 12),即将前两个维度保持不变,将最后一个维度展平为一个维度。
原文地址: http://www.cveoy.top/t/topic/cEJ4 著作权归作者所有。请勿转载和采集!