tensorflow.reduce_函数学习
@tensorflow.reduce_函数
tensorflow.reduce_sum()
reduce_sum函数的计算和指定维度的顺序无关
a = np.random.randint(1, 9, size=(2,3,4))
print(a)
b1 = tf.reduce_sum(tf.convert_to_tensor(value=a),axis=[2,1],keepdims=True)
print(b1)
b2 = tf.reduce_sum(tf.convert_to_tensor(value
原创
2021-11-02 16:24:17 ·
175 阅读 ·
0 评论