2020
09-27
09-27
tensorflow中tf.reduce_mean函数的使用
tf.reduce_mean函数用于计算张量tensor沿着指定的数轴(tensor的某一维度)上的的平均值,主要用作降维或者计算tensor(图像)的平均值。reduce_mean(input_tensor,axis=None,keep_dims=False,name=None,reduction_indices=None)第一个参数input_tensor:输入的待降维的tensor;第二个参数axis:指定的轴,如果不指定,则计算所有元素的均值;第三个参数keep_dims:是否降维度,设置为True...
继续阅读 >