Max¶
Returns the maximum value of all elements in the input tensor, or values and indices along dim.
Abstract Signature:
Max(input: Tensor, dim: int, keepdim: bool = False)
JAX (Core)
API:
jax.numpy.maxStrategy: Plugin (topk_adapter)
Returns the maximum value of all elements in the input tensor, or values and indices along dim.
Abstract Signature:
Max(input: Tensor, dim: int, keepdim: bool = False)
jax.numpy.max