torch.stack（）和torch.cat（）函数之间有什么区别？

torch.stack（）和torch.cat（）函数之间有什么区别？

stack

沿着 新的维度 连接张量序列。

cat

在给 定维度上 连接给定序列张量的序列。

因此，如果

A

和

B

具有形状（3，4），

torch.cat([A, B], dim=0)

则将具有形状（6，4），

torch.stack([A,B], dim=0)

并将具有形状（2，3，4）。