tf.keras.ops.amax
Stay organized with collections
Save and categorize content based on your preferences.
Returns the maximum of an array or maximum value along an axis.
tf.keras.ops.amax(
x, axis=None, keepdims=False
)
Args |
|---|
x
|
Input tensor.
|
axis
|
Axis along which to compute the maximum.
By default (axis=None), find the maximum value in all the
dimensions of the input array.
|
keepdims
|
If True, axes which are reduced are left in the result as
dimensions that are broadcast to the size of the original
input tensor. Defaults to False.
|
Returns |
|---|
An array with the maximum value. If axis=None, the result is a scalar
value representing the maximum element in the entire array. If axis is
given, the result is an array with the maximum values along
the specified axis.
|
Examples:
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)
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-06-07 UTC.
[null,null,["Last updated 2024-06-07 UTC."],[],[],null,["# tf.keras.ops.amax\n\n\u003cbr /\u003e\n\n|-----------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/keras-team/keras/tree/v3.3.3/keras/src/ops/numpy.py#L321-L355) |\n\nReturns the maximum of an array or maximum value along an axis.\n\n#### View aliases\n\n\n**Main aliases**\n\n[`tf.keras.ops.numpy.amax`](https://www.tensorflow.org/api_docs/python/tf/keras/ops/amax)\n\n\u003cbr /\u003e\n\n tf.keras.ops.amax(\n x, axis=None, keepdims=False\n )\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|------------|----------------------------------------------------------------------------------------------------------------------------------------------------------|\n| `x` | Input tensor. |\n| `axis` | Axis along which to compute the maximum. By default (`axis=None`), find the maximum value in all the dimensions of the input array. |\n| `keepdims` | If `True`, axes which are reduced are left in the result as dimensions that are broadcast to the size of the original input tensor. Defaults to `False`. |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| An array with the maximum value. If `axis=None`, the result is a scalar value representing the maximum element in the entire array. If `axis` is given, the result is an array with the maximum values along the specified axis. ||\n\n\u003cbr /\u003e\n\n#### Examples:\n\n x = keras.ops.convert_to_tensor([[1, 3, 5], [2, 3, 6]])\n keras.ops.amax(x)\n array(6, dtype=int32)\n\n x = keras.ops.convert_to_tensor([[1, 6, 8], [1, 5, 2]])\n keras.ops.amax(x, axis=0)\n array([1, 6, 8], dtype=int32)\n\n x = keras.ops.convert_to_tensor([[1, 6, 8], [1, 5, 2]])\n keras.ops.amax(x, axis=1, keepdims=True)\n array([[8], [5]], dtype=int32)"]]
View source on GitHub