tf.nn.softmax

分类: 文章 • 2024-01-01 16:40:58

tf.nn.softmax

tf.nn.softmax(
    logits,
    axis=None,
    name=None,
    dim=None
)

默认在最后一个维度上做softmax