花費 101 ms
pytorch中的cat、stack、tranpose、permute、unsqeeze

Cat 對數據沿着某一維度進行拼接。cat后數據的總維數不變. 比如下面代碼對兩個2維tensor(分別為2*3,1*3)進行拼接,拼接完后變為3*3還是2維的tensor。 import tor ...

Wed Aug 01 08:25:00 CST 2018 1 24925

 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM