【问题标题】:Add new elements to Pytorch tensor向 Pytorch 张量添加新元素
【发布时间】:2022-08-09 23:03:51
【问题描述】:

我有一个看起来像这样的张量:

我想为每个内部张量添加 0。所以我们有类似[0.1111, 0.6667, 0]的东西。 任何建议将不胜感激。

  • 你可以使用 torch.cat
  • PyTorch 张量由连续的内存区域支持。增长一个很可能需要重新分配内存(更大的块)并复制旧数据。您可以使用例如torch.cat 来执行此操作,或者只需创建一个正确大小的新张量并复制旧张量。但是,如果您发现自己依赖于此,您可能需要重新考虑您的算法。

标签: pytorch tensor


【解决方案1】:

你可以做 :

z = torch.zeros((10,1))
torch.cat((your_tensor,z),1)

【讨论】:

    猜你喜欢
    • 2020-10-04
    • 1970-01-01
    • 2019-04-21
    • 2020-07-20
    • 2013-04-17
    • 2021-09-14
    • 2019-04-18
    • 1970-01-01
    • 2021-01-10
    相关资源
    最近更新 更多