Di tutorial pemula MNIST , ada pernyataannya
accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float"))
tf.cast
pada dasarnya mengubah jenis tensor objek, tetapi apa perbedaan antara tf.reduce_mean
dan np.mean
?
Ini dokumen tentang tf.reduce_mean
:
reduce_mean(input_tensor, reduction_indices=None, keep_dims=False, name=None)
input_tensor
: Tensor yang akan dikurangi. Harus memiliki tipe numerik.
reduction_indices
: Dimensi yang akan dikurangi. JikaNone
(defaut), kurangi semua dimensi.
# '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.]
Untuk vektor 1D, sepertinya np.mean == tf.reduce_mean
, tapi saya tidak mengerti apa yang terjadi di tf.reduce_mean(x, 1) ==> [1., 2.]
. tf.reduce_mean(x, 0) ==> [1.5, 1.5]
jenis masuk akal, karena rata-rata [1, 2]
dan [1, 2]
adalah [1.5, 1.5]
, tapi apa yang terjadi dengan tf.reduce_mean(x, 1)
?