torch tensor张量 扩充维度 复制维度 expand repeat 深拷贝 浅拷贝 问题探析, 对tensor的广播计算, 高维度tensor 与 低维度tensor的兼容
expand 扩充维度的作用import torcha = torch.ones([1,1,2,3])b = a.expand( [1,2,2,3])print(a,'\n',b)a += 2print(a,'\n',b)output:atensor([[[[1., 1., 1.], [1., 1., 1.]]]]) b tensor([[[[1., 1., 1.], [1., 1., 1.]], [[1., 1., 1
复制链接