gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-02 18:55:01 24 4
gpt4 key购买 nike

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

例如:

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

我已经尝试使用 torch.cat([x1, X)axisdim 提供不同的数字,但它总是说尺寸不匹配。

最佳答案

您还可以使用torch.hstack结合和 unsqueeze用于 reshape x1

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

关于python - 将张量向量附加到张量矩阵,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/66299739/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com