torch.stack()和torch.cat()函数有什么区别？
6 浏览
torch.stack()和torch.cat()函数有什么区别？
OpenAI的强化学习中，REINFORCE和actor-critic示例代码如下：
REINFORCE：
policy_loss = torch.cat(policy_loss).sum()
actor-critic：
loss = torch.stack(policy_losses).sum() + torch.stack(value_losses).sum()
这两个示例中，一个使用了torch.cat函数，另一个使用了torch.stack函数，用于类似的用例。
就我理解而言，文档中没有明确区分它们之间的区别。
我很想知道这两个函数之间的区别。