torch.cat 是 PyTorch 中用于沿指定维度连接张量的函数。以下是该函数的详细用法：

语法

torch.cat(tensors, dim=0, *, out=None)

参数说明

• tensors (sequence of Tensors): 要连接的张量序列。这些张量必须具有相同的形状（除了连接的维度）。
• dim (int, optional): 沿着哪个维度进行连接。默认值为 0。
• out (Tensor, optional): 输出张量。

返回

• 一个新的张量，是将输入张量沿指定维度连接后的结果。

示例