使用tf.reduce_mean()做avg_pool

reduce_mean( input_tensor, axis=None, keepdims=None, name=None, reduction_indices=None, keep_dims=None ) axis表示按那個維度進行取平均值,當axis參數爲list時,會按照list的值依次進行取平均,對於二維張量其取值有0和1,表示兩個維度;當
相關文章
相關標籤/搜索