tf.keras.ops.max

Return the maximum of a tensor or maximum along an axis.

x Input tensor.
axis Axis or axes along which to operate. By default, flattened input is used.
keepdims If this is set to True, the axes which are reduced are left in the result as dimensions with size one. Defaults to False.
initial The minimum value of an output element. Defaults to None.

Maximum of x.