tf.nn.softmax(x, axis)裏axis起什麼做用?

在處理多分類問題時,tf.nn.softmax(x, axis) 函數是必定要使用的,那麼這裏的 axis 到底有什麼用呢?html 二維數組 首先,咱們來看在二維數組中的狀況:python A = np.array([[1, 2, 3], [1, 2, 3], [1, 2, 3], [1, 2, 3]]) A = tf.cast(A, tf.float32) A <tf.Tensor: id=1
相關文章
相關標籤/搜索