标签： 【深度学习】PyTorch之Squeeze()和Unsqueeze()

1. unsqueeze) 该函数用来增加某个维度。在PyTorch中维度是从0开始的。 import torch a = torch.arange0, 9) printa) 结果： ten … Continue reading 【深度学习】PyTorch之Squeeze和Unsqueeze