반응형
Deep Learning & Machine Learning/강좌&예제 코드2018. 8. 30. 19:40Tensorflow 예제 - tf.reduce_mean 함수 사용법

저수준 API로 작성된 MNIST 코드에서 사용하는 tf.reduce_mean 함수에 대해 살펴봅니다 import tensorflow as tf x = tf.constant([[1., 3.], [2., 6.]]) sess = tf.Session() print(sess.run(x)) print(sess.run(tf.reduce_mean(x))) print(sess.run(tf.reduce_mean(x, 0))) print(sess.run(tf.reduce_mean(x, 1))) sess.close() [[1. 3.] [2. 6.]]3.0[1.5 4.5][2. 4.] tf.reduce_mean(x)처럼 두번째 인자를 적지 않은 경우 변수 x가 가리키는 배열 전체 원소의 합을 원소 개수로 나누어 계산합니다. ..

반응형
image