tf.keras.ops.logsumexp
Вычисляет логарифм суммы экспонент элементов тензора.
tf.keras.ops.logsumexp(
x, axis=None, keepdims=False
)
| Аргументы | |
|---|---|
x | Входной тензор. |
axis | Целое число или кортеж целых чисел, задающий ось или оси, вдоль которых вычисляется сумма. Если None, сумма вычисляется по всем элементам. Значение по умолчанию: None. |
keepdims | Логическое значение, указывающее, нужно ли сохранять размерности входного тензора при вычислении суммы. Значение по умолчанию: False. |
| Возвращает | |
|---|---|
Тензор, содержащий логарифм суммы экспонент элементов в x. |
Пример:
x = keras.ops.convert_to_tensor([1., 2., 3.]) logsumexp(x) 3.407606
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/logsumexp