相关疑难解决方法(0)

torch.stack()和torch.cat()函数有什么区别?

OpenAI的REINFORCE和演员批评强化学习的例子有以下代码:

加强:

policy_loss = torch.cat(policy_loss).sum()
Run Code Online (Sandbox Code Playgroud)

演员评论家:

loss = torch.stack(policy_losses).sum() + torch.stack(value_losses).sum()
Run Code Online (Sandbox Code Playgroud)

一个是使用torch.cat,另一个是使用torch.stack.

就我的理解而言,该文件并未对它们作出任何明确的区分.

我很高兴知道这些功能之间的差异.

python machine-learning pytorch

16
推荐指数
2
解决办法
8861
查看次数

标签 统计

machine-learning ×1

python ×1

pytorch ×1