python中torch.cat函数的用法

torch.cat函数用于将多个张量沿着指定维度进行拼接。它的语法为：torch.cat(tensors, dim=0, out=None)。其中，tensors是要拼接的张量序列，dim是拼接的维度，out是输出张量。例如，torch.cat([x, y, z], dim=0)会将三个张量x、y、z在第0维进行拼接。