如果你有两个形状为 (3, 4)
的tensor A
和 B
,并且你想要在第一个维度上连接它们,你可以这样做:
import torchA = 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
的元素数量。