使用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,表示两个维度;当
相关文章
相关标签/搜索