pytorch中expand函数的使用
信息源地址:torch.Tensor - PyTorch中文文档1.理论解释2.实践与理解expand1.tensor.expand()函数可以将维度值包含 1 的Tensor（如：torch.Size([1, n])或者torch.Size([n, 1])）的维度进行扩展2.将 -1 传递给新扩展维度或者无需扩展维度均表示不更改该维度的尺寸。2. expand_as:同expand(x.shap
·
信息源地址: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后相乘的操作
有“AI”的1024 = 2048，欢迎大家加入2048 AI社区
更多推荐
14
0- 0
已为社区贡献3条内容
所有评论(0)