size()
size()函数返回张量的各个维度的尺度。
squeeze()
squeeze(input, dim=None),如果不给定dim,则把input的所有size为1的维度给移除;如果给定dim,则只移除给定的且size为1的维度。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:pytorch中的size()、 squeeze()函数 - Python技术站
size()函数返回张量的各个维度的尺度。
squeeze(input, dim=None),如果不给定dim,则把input的所有size为1的维度给移除;如果给定dim,则只移除给定的且size为1的维度。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:pytorch中的size()、 squeeze()函数 - Python技术站