zl程序教程

您现在的位置是:首页 >  后端

当前栏目

【PyTorch】 tensor.unsqueeze扩展一个维度

扩展PyTorch 一个 维度 Tensor
2023-09-14 09:12:47 时间

【PyTorch】 tensor.unsqueeze扩展一个维度

1、背景

在tensor合并的时候,经常需要考虑tensor维度的对齐,即涉及维度的扩展;
pytorch中关于维度扩展的方法:unsqueeze(…)即解压缩(扩展)

关于unsqueeze方法的说明如下:

import torch
print(torch.__version__)
help(torch.unsqueeze)

运行结果如下:

Help on built-in function uns