qq_44582785的博客

原创 tf.reduce_mean()的用法

tf.math.reduce_mean( input_tensor, axis=None, keepdims=False, name=None ) 作用：按给定维度实现平均，将指定维度大小变为1 参数：input_tensor输入张量，axis变换的维度（求和后平均）(默认值为所有元素求和取平均)，keepdims输出结果是否保持维度（默认为不保持原维度） 代码示例： import tensorflow as tf x = tf.constant([[[1., 1., 1], [2., 2....