def argmax(module, x, axes=None):
    """
    Get indices of maximum in tensor.

    Arguments:
        module: The backend array corresponding to the given array.
        x: The tensor to calculate the exponential of.
        axes: Tuple specifying the axes along the which compute the
            maximum.

    Return:
        Tensor containing indices of the maximum in 'x'.
    """
    return module.argmax(x, axes)