pytorch中pixelshuffle的具體操作方式

官方文檔中對於pixelshuffle的解釋:https://pytorch.org/docs/0.3.1/_modules/torch/nn/modules/pixelshuffle.html 簡單點說就是一個調整數據順序的過程,即輸入的張量大小爲:, 調整爲: 的過程,可是官方文檔只說了這些,並沒有解釋怎樣對應起來的。對應公式如下, , a爲輸入的tensor, 分別是batch, chane
相關文章
相關標籤/搜索