max

Graph.max(x, axis=None, keepdims=False, *, name=None)

Find the maximum value in a tensor along one or multiple of its axes.

Parameters

• x (np.ndarray or Tensor) – The tensor of which you want to find the maximum value.

• axis (int or list[int] or tuple[int], optional) – The dimension or dimensions along which you want to search the tensor. Defaults to None, in which case this node returns the maximum value along all axes of the tensor.

• keepdims (bool, optional) – Whether or not to retain summed axes in the output tensor. If True, each dimension in axis has size 1 in the result; otherwise, the dimensions in axis are removed from the result. Defaults to False.

• name (str, optional) – The name of the node.

Returns

The maximum tensor obtained by searching the input tensor along the specified axes.

Return type

Tensor

See also

min

Find the minimum value in a tensor along one or multiple of its axes.