torch.stack()関数とtorch.cat()関数の違いは何ですか?

ブログ

torch.stack()関数とtorch.cat()関数の違いは何ですか?

OpenAIのREINFORCEと強化学習のアクター批評の例には、次のコードがあります。

強化する

policy_loss = torch.cat(policy_loss).sum() 

俳優批評家

loss = torch.stack(policy_losses).sum() + torch.stack(value_losses).sum() 

1つはtorch.catを使用し、もう1つはtorch.stackを使用します。

私の理解では、 ドキュメント それらの間に明確な区別はありません。

機能の違いを教えていただければ幸いです。

#python#machine-learning