python-3.x 如何用一个Tensor扩展另一个Tensor,因此结果包含了2个Tensor的所有元素，见示例

ktca8awb 于 2023-01-14 发布在 Python

关注(0)|答案(1)|浏览(164)

a =  tensor([   [101,  103],
            [101,  1045]
        ])

b =  tensor([   [101,  777],
            [101,  888]
        ])

如何从a和B得到Tensorc：

c = a + b =  tensor([   [101,  103],
            [101,  1045],
            [101,  777],
            [101,  888]
            
        ])

对于python列表，这将是简单的c = a + b，但对于pytorch，它只是简单地添加元素，而不是扩展列表。

python-3.x

来源：https://stackoverflow.com/questions/75103799/how-to-extend-one-tensor-with-another-so-the-result-contains-all-the-elements-f

1条答案

按热度按时间

6yoyoihd1#

您可以使用torch.cat函数：

c = torch.cat((a, b), dim=0)

如以下示例所示：

from torch import tensor
import torch
a =  tensor([   [101,  103],
            [101,  1045]
        ])

b =  tensor([   [101,  777],
            [101,  888]
        ])
c = torch.cat((a, b), dim=0)
print(c)

输出：

tensor([[ 101,  103],
        [ 101, 1045],
        [ 101,  777],
        [ 101,  888]])

赞(0）回复(0）举报 2023-01-14

我来回答

python-3.x 如何用一个Tensor扩展另一个Tensor,因此结果包含了2个Tensor的所有元素，见示例

1条答案

相关问题

热门标签

最新问答