深度學習與PyTorch筆記7

Tensor維度變換 view/reshape lost dim information view和reshape用法完全相同,將數據合併,必須要有物理意義。適用於全連接層。缺點:不好還原數據。 squeeze/unsqueeze squeeze:擠壓、減少維度 unsqueeze:展開、增加維度,取值範圍爲[-a.dim()-1,a.dim()+1) 正數:在前面插入。負數:在後面插入。 exp
相關文章
相關標籤/搜索