早上好
我正在搜索复制张量的方法。
例如,
a = torch.Tensor(1, 4) ## tensor([[0.1, 0.2, 0.3, 0.4]])
我要像这样将其变成4x4张量
tensor([[0.1, 0.1, 0.1, 0.1],
[0.2, 0.2, 0.2, 0.2],
[0.3, 0.3, 0.3, 0.3],
[0.4, 0.4, 0.4, 0.4]])
进行张量为4xN的乘法
有保持梯度的想法吗? (我不完全了解如何保持每个张量的梯度。)
谢谢