还有一种形式就是b=torch.squeeze(a,N) a中去掉指定的定的维数为一的维度。 再看torch.unsqueeze()这个函数主要是对数据维度进行扩充。给指定位置加上维 ...
確定! 回上一頁