tf.nn.softmax的axis的理解

import tensorflow as tf
tf.enable_eager_execution()
ones = tf.ones(shape=[2,3])
temp1 = tf.nn.softmax(ones,axis=0) # 列
print(temp1)
temp2 = tf.nn.softmax(ones,axis=1) # 行
print(temp2)

打印结果:

tf.Tensor(
[[0.5 0.5 0.5]
[0.5 0.5 0.5]], shape=(2, 3), dtype=float32)
tf.Tensor(
[[0.33333334 0.33333334 0.33333334]
[0.33333334 0.33333334 0.33333334]], shape=(2, 3), dtype=float32)

你可能感兴趣的:(TensorFlow)