tf.keras.ops.amax
Возвращает максимальное значение массива или максимальное значение по оси.
tf.keras.ops.amax(
x, axis=None, keepdims=False
)
| Аргументы | |
|---|---|
x | Входной тензор. |
axis | Ось, по которой вычисляется максимум. По умолчанию (axis=None), находится максимальное значение во всех измерениях входного массива. |
keepdims | Если True, оси, которые были уменьшены, остаются в результате как размеры, которые транслируются до размера исходного тензора. По умолчанию False. |
| Возвращаемое значение | |
|---|---|
Массив с максимальным значением. Если axis=None, результат представляет собой скалярное значение, представляющее максимальный элемент во всем массиве. Если указан axis, результат представляет собой массив с максимальными значениями по указанной оси. |
Примеры:
x = keras.ops.convert_to_tensor([[1, 3, 5], [2, 3, 6]]) keras.ops.amax(x) array(6, dtype=int32)
x = keras.ops.convert_to_tensor([[1, 6, 8], [1, 5, 2]]) keras.ops.amax(x, axis=0) array([1, 6, 8], dtype=int32)
x = keras.ops.convert_to_tensor([[1, 6, 8], [1, 5, 2]]) keras.ops.amax(x, axis=1, keepdims=True) array([[8], [5]], dtype=int32)
© 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/amax