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

size() size()函數返回張量的各個維度的尺度。 squeeze() squeeze(input, dim=None)，如果不給定dim，則把input的所有size為1的維度給移除；如果給定dim，則只移除給定的且size為1的維度。 ...