信息源地址:torch.Tensor - PyTorch中文文档
1.理论解释
2.实践与理解
expand
1.tensor.expand()函数
可以将维度值包含 1 的Tensor(如:torch.Size([1, n])
或者torch.Size([n, 1])
)的维度进行扩展
2.将 -1 传递给新扩展维度或者无需扩展维度均表示不更改该维度的尺寸。
2. expand_as:同expand(x.shape)用法
结论:关于expand用法,注意到只能将维度值包含 1 的Tensor(如:torch.Size([1, n])
或者torch.Size([n, 1])
)的维度进行扩展,否则会报错
对应的维度为1可以理解为CNN中GAP后相乘的操作