tf.contrib.layers.softmax
Выполняет softmax по N-й размерности N-мерного тензора логарифмированных вероятностей.
tf.contrib.layers.softmax(
logits, scope=None
)
Для двумерных логарифмов это сводится к tf.nn.softmax. N-й размер должен иметь заданное число элементов (количество классов).
| Аргументы | |
|---|---|
logits | N-мерный Tensor с логарифмами, где N > 1. |
scope | Необязательный scope для variable_scope. |
| Возвращает | |
|---|---|
A Tensor с такой же формой и типом, как логарифмы. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/layers/softmax
