torch.cat 是 PyTorch 中用于沿指定維度連接張量的函數。以下是該函數的詳細用法：

語法

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

參數說明

• tensors (sequence of Tensors): 要連接的張量序列。這些張量必須具有相同的形狀（除了連接的維度）。
• dim (int, optional): 沿著哪個維度進行連接。默認值為 0。
• out (Tensor, optional): 輸出張量。

返回

• 一個新的張量，是將輸入張量沿指定維度連接后的結果。

示例