Torch NDArray .max(int, boolean) should return tuple.

Currently NDArray.max(int, boolean) only returns the max values. However it should also give the indexes in a tuple as per pytorch docs.

https://pytorch.org/docs/stable/generated/torch.max.html
"Returns a namedtuple (values, indices) where values is the maximum value of each row of the input tensor in the given dimension dim. And indices is the index location of each maximum value found (argmax)."

If djl's version of .max() on NDArray is intentional, please share what is the best DJL way to calculate the index of each max value. Thank you!