Trong hướng dẫn cho người mới bắt đầu MNIST , có câu
accuracy = tf.reduce_mean(tf.cast(correct_prediction, "float"))
tf.cast
về cơ bản thay đổi loại tensor của đối tượng, nhưng sự khác biệt giữa tf.reduce_mean
và là np.mean
gì?
Đây là tài liệu về tf.reduce_mean
:
reduce_mean(input_tensor, reduction_indices=None, keep_dims=False, name=None)
input_tensor
: Tensor để giảm. Nên có kiểu số.
reduction_indices
: Các kích thước cần giảm. NếuNone
(giá trị mặc định), giảm tất cả các thứ nguyên.# '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.]
Đối với vectơ 1D, nó trông giống như vậy np.mean == tf.reduce_mean
, nhưng tôi không hiểu điều gì đang xảy ra tf.reduce_mean(x, 1) ==> [1., 2.]
. tf.reduce_mean(x, 0) ==> [1.5, 1.5]
loại có ý nghĩa, vì có nghĩa là [1, 2]
và [1, 2]
hiện tại [1.5, 1.5]
, nhưng điều gì đang xảy ra với tf.reduce_mean(x, 1)
?