pytorch 入門(七):unsqueeze. ygys1234 發表於2020-11-14. PyTorch ... 用squeeze做升維操作,squeeze(n)的引數n指定新維度的位置。 print(a.unsqueeze(0).shape) ...
確定! 回上一頁