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

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