torch.cat

torch.cat() 是 PyTorch 中的一个函数，用于将张量沿着指定的维度进行拼接。它的语法如下：

torch.cat(tensors, dim=0, out=None) -> Tensor

其中，tensors 是要拼接的张量序列，dim 是拼接的维度，out 是拼接结果的输出张量（可选参数）。

例如，假设有两个形状为 (2, 3) 的张量 a 和 b，我们可以使用 torch.cat() 将它们沿着第 0 维拼接成一个形状为 (4, 3) 的张量：

import torch

a = torch.randn(2, 3)
b = torch.randn(2, 3)
c = torch.cat([a, b], dim=0)
print(c.shape)  # 输出：(4, 3)

注意，要求拼接的张量在除了指定维度以外的其他维度上大小相同，否则会抛出异常。