torch.cat与torch.stack的区别

1.torch.cat

语法:

• torch.cat([A1,A2,A3,...],dim=)
  将多个张量按照指定维度拼接起来

• 用法示例：
  

2. torch.stack

语法:

• torch.stack([A1,A2,A3,...],dim=)
  新增一个维度，沿着一个新维度对输入张量序列进行连接。 序列中所有的张量都应该为相同形状
• 用法示例：