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