Qual è la differenza tra le funzioni torch.stack() e torch.cat()?

Blog

Qual è la differenza tra le funzioni torch.stack() e torch.cat()?

L'esempio REINFORCE di OpenAI e l'attore critico per l'apprendimento per rinforzo ha il seguente codice:



RAFFORZARE :

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

attore-critico :



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

Uno usa torch.cat, l'altro usa torch.stack.

Per quanto la mia comprensione va, il documento non fornisce alcuna chiara distinzione tra loro.



Sarei felice di conoscere le differenze tra le funzioni.

#python #machine-learning