花费 58 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