您的位置: 首页 > 文章 > torch.Tensor.expand详解 torch.Tensor.expand详解 分类: 文章 • 2022-10-27 23:26:08 torch.Tensor.expand详解 expand()函数可以将张量广播到新的形状,但是切记以下两点: 1. 只能对维度值为1的维度进行扩展,且扩展的Tensor不会分配新的内存,只是原来的基础上创建新的视图并返回; 2. 无需扩展的维度请保持维度值不变。