保持梯度的pytorch张量复制

时间:2020-07-04 04:19:49

标签: duplicates pytorch gradient tensor torch

早上好
我正在搜索复制张量的方法。
例如,

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的乘法
有保持梯度的想法吗? (我不完全了解如何保持每个张量的梯度。)

谢谢

0 个答案:

没有答案