Quali sono i modi più idiomatici per combinare i tensori nella torcia?

Question

Mi trovo di fronte a concatenare tre tensioni insieme in modo che i tensore 3 px1 diventino un tensore 3px1.

Il più succinta ho potuto venire con era:

torch.Tensor{v2:totable(),v4:totable(),v6:totable()}:view(3*p,1) 

ci sono modi per fare questo senza la conversione in tabelle e di nuovo a tensori? Sembra che ci dovrebbe essere un modo generico di concatenare i tensori lungo una determinata dimensione assumendo che abbiano forme compatibili.

Posso vedere come sarebbe possibile scrivere una funzione del genere, non esiste?

Answer 1

a = torch.randn(3,1) 
b = torch.randn(3,1) 
c = torch.randn(3,1) 

d = torch.cat(a,b,1):cat(c,1) 

print(d)