pytorch中的size()、 squeeze()函数

size()

size()函数返回张量的各个维度的尺度。

squeeze()

squeeze(input, dim=None)，如果不给定dim，则把input的所有size为1的维度给移除；如果给定dim，则只移除给定的维度中size为1的维度。