代码之家  ›  专栏  ›  技术社区  ›  Stepan Yakovenko

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

  •  0
  • Stepan Yakovenko  · 技术社区  · 6 年前

    创建wxw张量:

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

    做一些计算:

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

    如何从x u upd分配x数据?

    2 回复  |  直到 6 年前
        1
  •  0
  •   llllllllll    6 年前
        2
  •  0
  •   Stepan Yakovenko    6 年前

    好吧,解决办法是:

    x.data.from_numpy(x_upd)