§
    `ƒi«  ã            	       óX   — d dl Zej                             ddddddd¬	¦  «        Zdd
„ZdS )é    NzT x1zT yzexp(x1)za + bz
y = log(a)Ú0Úlog_softmax)Únamec                 óª  — t          j        | |d¬¦  «        }|j        dk    rd|t          j        |¦  «         <   nt          j        |¦  «        sd}| |z
  }|j        j        dv rWt           j        t           j        t           j        fD ]3}t          j	        |j        |¦  «        r| 
                    |¦  «        } nŒ4t          ||d¬¦  «        }||z
  }|S )aæ  Compute logarithm of softmax function

    Parameters
    ----------
    x : array-like
        Input array
    axis : int or tuple of ints, optional
        Axis to compute values along. Default is None and softmax
        will be  computed over the entire array `x`

    Returns
    -------
    s : cupy.ndarry
        An array with the same shape as `x`. Exponential of the
        result will sum to 1 along the specified axis. If `x` is a
        scalar, a scalar is returned

    T)ÚaxisÚkeepdimsr   Úiu)ÚcpÚamaxÚndimÚisfiniteÚdtypeÚkindÚfloat16Úfloat32Úfloat64Úcan_castÚastypeÚ_log_softmax_kernel)Úxr   Úx_maxÚtmpÚ	out_dtypeÚouts         ús/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupyx/scipy/special/_logsoftmax.pyr   r      sß   € õ( ŒGA˜D¨4Ð0Ñ0Ô0€Eà„zA‚~€~Ø%&ˆrŒ{˜5Ñ!Ô!Ð!Ñ"Ð"ÝŒ[˜ÑÔð Øˆà
ˆe‰)€Cà
„y„~˜ÐÐÝœ*¥b¤jµ"´*Ð=ð 	ð 	ˆIÝŒ{˜3œ9 iÑ0Ô0ð Ø—j’j Ñ+Ô+Øðõ ˜c¨°tÐ
<Ñ
<Ô
<€Cà
‰)€CØ€Jó    )N)Úcupyr
   Ú_coreÚReductionKernelr   r   © r   r   ú<module>r!      sc   ðØ Ð Ð Ð ð ”h×.Ò.Ø
Ø	ØØØØØ	ð /ñ ô Ð ð&ð &ð &ð &ð &ð &r   