tensorflow reduction_indices函數

在tensorflow的使用中,經常會使用tf.reduce_mean,tf.reduce_sum等函數,在函數中,有一個reduction_indices參數,表示函數的處理維度,直接上圖,一目瞭然: 需要注意的一點,在很多的時候,我們看到別人的代碼中並沒有reduction_indices這個參數,此時該參數取默認值None,將把input_tensor降到0維,也就是一個數。
相關文章
相關標籤/搜索