torch squeeze 压缩维度