Pytorch中torch.cat()函数的使用及说明

如果你有两个形状为 (3, 4) 的tensor A 和 B，并且你想要在第一个维度上连接它们，你可以这样做：

import torch
 
A = torch.randn(3, 4)
B = torch.randn(3, 4)
C = torch.cat([A, B], dim=0)  # dim=0 表示在第一个维度上连接

此时 C 的形状将会是 (6, 4)，因为你在第一个维度上加入了 A 和 B 的元素数量。

C = torch.cat([A, B], dim=1)  # dim=1 表示在第二个维度上连接

此时 C 的形状将会是 (3, 8)，因为在第二个维度上加入了 A 和 B 的元素数量。