torch.Tensor.expand详解expand()函数可以将张量广播到新的形状,但是切记以下两点:1. 只能对维度值为1的维度进行扩展,且扩展的Tensor不会分配新的内存,只是原来的 ...
確定! 回上一頁