code: def reduce_max(x,
                 disable_positional_args=None,
                 output_shape=None,
                 reduced_dim=None,
                 name=None):
    
    output_shape = convert_to_shape(output_shape)
    reduced_dim = convert_to_dimension(reduced_dim)
    assert disable_positional_args is None
    output_shape = _reduction_output_shape(x, output_shape, reduced_dim)
    if output_shape is None:
      output_shape = Shape([])
    if output_shape == x.shape:
      return x
    return ReduceOperation(
        x, output_shape, "MAX", name=name or "reduce_max").outputs[0]
docs: Reduction on 1 or more axes.
  
    Args:
      x: a Tensor
      disable_positional_args: None
      output_shape: an optional Shape.  Must be a subsequence of x.shape.
      reduced_dim: an optional Dimension
      name: an optional string
    Returns:
      a Tensor