에서 MNIST 초보자 튜토리얼 , 문이있다
accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float"))
tf.cast
기본적으로 객체의 텐서 유형을 변경하지만 tf.reduce_mean
과 의 차이점은 무엇 np.mean
입니까?
에 대한 문서는 다음과 같습니다 tf.reduce_mean
.
reduce_mean(input_tensor, reduction_indices=None, keep_dims=False, name=None)
input_tensor
: 줄일 텐서입니다. 숫자 유형이어야합니다.
reduction_indices
: 줄일 치수입니다. (기본) 경우None
모든 치수를 줄입니다.
# 'x' is [[1., 1. ]] # [2., 2.]] tf.reduce_mean(x) ==> 1.5 tf.reduce_mean(x, 0) ==> [1.5, 1.5] tf.reduce_mean(x, 1) ==> [1., 2.]
1D 벡터의 경우처럼 보이지만 np.mean == tf.reduce_mean
에서 무슨 일이 일어나고 있는지 이해하지 못합니다 tf.reduce_mean(x, 1) ==> [1., 2.]
. tf.reduce_mean(x, 0) ==> [1.5, 1.5]
의 mean [1, 2]
과 [1, 2]
is 이므로 의미 가 있습니다 [1.5, 1.5]
.하지만 무슨 일이 일어나고 tf.reduce_mean(x, 1)
있습니까?