stack
沿着 新的维度 连接张量序列。
cat
在给 定维度上 连接给定序列张量的序列。
因此，如果
A和
B具有形状（3，4），
torch.cat([A, B], dim=0)则将具有形状（6，4），
torch.stack([A,B], dim=0)并将具有形状（2，3，4）。
stack
沿着 新的维度 连接张量序列。
cat
在给 定维度上 连接给定序列张量的序列。
因此，如果
A和
B具有形状（3，4），
torch.cat([A, B], dim=0)则将具有形状（6，4），
torch.stack([A,B], dim=0)并将具有形状（2，3，4）。
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
