深度學習與PyTorch筆記8

Broadcast自動擴展 作用:expand維度擴展without copying data不需要複製數據。 步驟: 1、insert 1 dim ahead,最前面自動擴展一個維度,shape爲1。 2、expand dim with size 1 to same size,將維度爲1的擴展成與原數據相同的大小 feature maps:[4,32,14,14] bias:[32,1,1]–[
相關文章
相關標籤/搜索