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
相关文章
相关标签/搜索