2 . AlexNet(2)

x 輸入數據 數據種類 (爲2 因爲只包含貓狗) keep_prob 也就是dropout層的參數 skip layer 爲不需要使用伯克利初始化的層 。 權重位置參數 最後是網絡創建函數create的調用 下圖是 create 函數調用情況 pool5後 把得到的 進行 reshape函數 變成一個向量 下面是conv 函數實現細節 首先獲取最後一個維度也就是通道數 下面是 匿名函數 對i k
相關文章
相關標籤/搜索