pytorch - 在pytorch中合并两个张量

Question

张量a:

tensor([[1, 2, 3],
        [1, 2, 3],
        [1, 2, 3],
        [1, 2, 3]])

张量 b:

tensor([4,4,4,4])

问题1:

如何合并两个张量并得到结果c:

tensor([[1, 2, 3, 4],
        [1, 2, 3, 4],
        [1, 2, 3, 4],
        [1, 2, 3, 4]])

问题2:如何将张量c除以得到原始的a和b。

Answer 1

问题 1:合并两个张量 -

torch.cat((a, b.unsqueeze(1)), 1)
>>> tensor([[1, 2, 3, 4],
            [1, 2, 3, 4],
            [1, 2, 3, 4],
            [1, 2, 3, 4]])

首先，我们使用torch.unsqueeze在b张量中添加单个dim以匹配要连接的adim。然后使用torch.cat连接张量 a 和 b。

问题2:

a = c[:][:,:-1]
a
>>> tensor([[1, 2, 3],
            [1, 2, 3],
            [1, 2, 3],
            [1, 2, 3]])

b = c[:][:,-1:].squeeze(1)
b
>>> tensor([4, 4, 4, 4])