size()函数返回张量的各个维度的尺度。
squeeze(input, dim=None)，如果不给定dim，则把input的所有size为1的维度给移除；如果给定dim，则只移除给定的维度中size为1的维度。
pytorch中的size()、 squeeze()函数
原文：https://www.cnblogs.com/picassooo/p/12528834.html
