TensorFlow基礎1:reduce_sum()函數和reduce_mean()函數

在計算損失時,通常會用到reduce_sum()函數來進行求和,但是在使用過程中常常會搞不清楚具體是怎樣進行計算的,通過查閱資料,逐漸搞清楚了這個函數的用法,下面就來詳細解釋一下。 在TensorFlow官方文檔中有這樣的解釋: 其實在reduce_sum()中,是從維度上去考慮的。其中的參數reduction_indices很容易搞蒙圈,上個圖加深理解吧。 調用reduce_sum(arg1,
相關文章
相關標籤/搜索