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