TensorFlow函数教程:tf.nn.softmax

由 Carrie 创建, 最后一次修改 2019-02-01

tf.nn.softmax函数

别名:

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

定义在:tensorflow/python/ops/nn_ops.py。

计算softmax激活。(弃用的参数)

有些参数已被弃用。它们将在将来的版本中删除。更新说明:不推荐使用dim,而是使用axis。

此函数执行相当于:

softmax = tf.exp(logits) / tf.reduce_sum(tf.exp(logits), axis)

参数:

  • logits:一个非空的Tensor。必须是下列类型之一:half, float32,float64。
  • axis:将在其上执行维度softmax。默认值为-1,表示最后一个维度。
  • name:操作的名称(可选)。
  • dim:axis的已弃用的别名。

返回:

一个Tensor,与logits具有相同的类型和shape。

可能引发的异常:

  • InvalidArgumentError:如果logits为空或axis超出logits的最后一个维度。
以上内容是否对您有帮助:
二维码
建议反馈
二维码