使用numpy.expand_dims可以将numpy.ndarray数据增加一个维度,数值为1,与np.squeeze相反,类似于pytorch中的torch.squeeze和torch.unsqueezenumpy.expand_dims(a, ...
確定! 回上一頁