在tensorflow的使用中,经常会使用tf.reduce_mean,tf.reduce_sum等函数,在函数中,有一个reduction_indices参数,表示函数的处理维度,直接上图,一目了然:

博客介绍了在TensorFlow使用中,tf.reduce_mean、tf.reduce_sum等函数里的reduction_indices参数,该参数表示函数处理维度。还指出若代码中未出现此参数,其默认值为None,会将input_tensor降到0维成为一个数。

您可能感兴趣的与本文相关的镜像
TensorFlow-v2.15
TensorFlow 是由Google Brain 团队开发的开源机器学习框架,广泛应用于深度学习研究和生产环境。 它提供了一个灵活的平台,用于构建和训练各种机器学习模型

被折叠的 条评论
为什么被折叠?