pytorch中expand函数的使用

信息源地址: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后相乘的操作