【问题标题】:Torch: Concatenating tensors of different dimensionsTorch:连接不同维度的张量
【发布时间】:2017-09-22 17:36:00
【问题描述】:

我有一个x_at_i = torch.Tensor(1,i),它在 i = 0 到 n 的每次迭代中都会增长。我想将所有不同大小的张量连接成一个矩阵,并用零填充剩余的单元格。什么是最惯用的方法。例如:

x_at_1 = 1
x_at_2 = 1 2
x_at_3 = 1 2 3
x_at_4 = 1 2 3 4
X = torch.cat(x_at_1, x_at_2, x_at_3, x_at_4)
X = [ 1 0 0 0
      1 2 0 0
      1 2 3 0
      1 2 3 4 ]

【问题讨论】:

    标签: lua size concatenation torch tensor


    【解决方案1】:

    如果您知道 n 并假设您可以在每次迭代中轻松访问您的 x_at_i,我会尝试类似

    X = torch.Tensor(n, n):zero()
    for i = 1, n do
        X[i]:narrow(1, 1, i):copy(x_at[i])
    end
    

    【讨论】:

      猜你喜欢
      • 2017-09-30
      • 2022-01-22
      • 2020-09-24
      • 2018-10-29
      • 1970-01-01
      • 2021-12-09
      • 2021-04-04
      • 2016-05-02
      • 1970-01-01
      相关资源
      最近更新 更多