size()

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

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

squeeze()

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

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

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