quantnn/generic/__init__.py
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)

    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'.
    
Get indices of maximum in tensor.
Get indices of maximum in tensor .
def argmax(module, x, axes=None):
    return module.argmax(x, axes)
def argmax ( module , x , axes = None ) : return module . argmax ( x , axes )
Get indices of maximum in tensor.
Get indices of maximum in tensor .
"""
    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'.
    """
module
The backend array corresponding to the given array.
The backend array corresponding to the given array .
x
The tensor to calculate the exponential of.
The tensor to calculate the exponential of .
axes
Tuple specifying the axes along the which compute the
maximum.
Tuple specifying the axes along the which compute the maximum .