如何从 numpy 为pytorch张量分配矩阵?

Question

Create WxW tensor：

x = Variable(torch.FloatTensor(W,W).zero_(), requires_grad=True)

做一些计算：

x_copy = x0=np.copy(x.data.numpy())
x_upd = handleArray(x_copy)

如何从x_upd分配 x 数据？

Answer 1

考虑使用tensor.from_numpy()

https://pytorch.org/tutorials/beginner/former_torchies/tensor_tutorial.html

Answer 2

好的，解决方案是执行以下操作：

x.data.from_numpy(x_upd)