【PyTorch】 tensor.unsqueeze扩展一个维度
1、背景
在tensor合并的时候,经常需要考虑tensor维度的对齐,即涉及维度的扩展;
pytorch中关于维度扩展的方法:unsqueeze(…)即解压缩(扩展)
关于unsqueeze方法的说明如下:
import torch
print(torch.__version__)
help(torch.unsqueeze)
运行结果如下:
Help on built-in function unsqueeze:
unsqueeze(