python - 将张量向量附加到张量矩阵

Question

我有一个张量矩阵，我只想将张量向量作为另一列附加到它。

例如：

    X = torch.randint(100, (100,5))
    x1 = torch.from_numpy(np.array(range(0, 100)))

我已经尝试torch.cat([x1, X)了两种不同的数字axis，dim但它总是说尺寸不匹配。

标签： pythonpytorch

Answer 1

您还可以使用torch.hstackto combine 和unsqueezefor reshapex1

torch.hstack([X, x1.unsqueeze(1)])