+
    LV-j…K  ã                   óº  € ^ RI t^ RIt^ RIt^ RIt^ RIHtHtH	t	H
t
 ^ RIHtHtHt ^ RIHtHtHt ^ RIHt . ROtRR.tR t]P0                  P3                  R	4      ]P0                  P3                  R
4      ]P0                  P3                  R4      ]! ]4       ! R R4      4       4       4       4       t]! ]4       ! R R4      4       t]! ]4       ! R R4      4       tR# )é    N)Úis_array_api_strictÚmake_xp_test_caseÚxp_default_dtypeÚ	xp_device)Úxp_assert_equalÚxp_assert_closeÚxp_assert_less)Úlog_softmaxÚ	logsumexpÚsoftmax)Ú_wrap_radiansÚint32Úint64c                 ó¾  € V P                  \        P                  ) ^,
          \        P                  ) RR^ R^\        P                  \        P                  ^,           .	4      pV P                  \        P                  ^,
          \        P                  RR^ R^\        P                  \        P                  ) ^,           .	4      p\        WR7      p\	        W2^ R7       R# )é   gYóøÂn¥)Úxp)ÚatolNéÿÿÿÿgYóøÂn¥�)ÚasarrayÚmathÚpir   r   )r   ÚxÚrefÚress   &   Ús/Volumes/fast/ai/experiments/ui-tars-smoke/.venv/lib/python3.14/site-packages/scipy/special/tests/test_logsumexp.pyÚtest_wrap_radiansr      s˜   € Ø
�
‰
”T—W‘W�H˜Q•J¤§¡ ¨"¨gØ�v˜q¤$§'¡'¬4¯7©7°1­9ð6ó 	7€Aà
�*‰*”d—g‘g˜a•i¤§¡¨"¨gØ�v˜q¤$§'¡'¬D¯G©G¨8°A­:ð7ó 8€Cä
˜Ô
!€CÜ�C 1×%ó    z/ignore:invalid value encountered:RuntimeWarningz0ignore:divide by zero encountered:RuntimeWarningz*ignore:overflow encountered:RuntimeWarningc                   óÔ  a € ] tR t^ t o R tR tR tR tR tR t	R t
R tR	 t]P                  P                  R
^. R)O34      R 4       tR t]P                  P                  R]4      R 4       t]P                  P                  R]4      ]P                  P                  R]4      R 4       4       tR t]P                  P                  RRR.4      R 4       tR t]P                  P                  R]P2                  ! ]P6                  ) ]P6                  ]! ]P6                  ) R4      ]! ]P6                  ) R*4      ]! ]P6                  ) ]P6                  4      ]! ]P6                  ) ]P6                  ) 4      ]! ]P6                  R4      ]! ]P6                  R*4      ]! ]P6                  ]P6                  4      ]! ]P6                  ]P6                  ) 4      ]! ]P6                  ) R4      ]! ]P6                  ) R4      ]! ]P6                  ) R4      ]! ]P6                  ) R4      ]! ]P6                  R4      ]! ]P6                  R4      ]! ]P6                  R4      ]! ]P6                  R4      .^R7      4      R 4       tR t]P                  P                  R R!R". .4      R# 4       tR$ t ]P                  P                  R%]P6                  .R&R!3]P6                  .R+.R+3.4      R' 4       t!R(t"V t#R&# ),ÚTestLogSumExpc           	     ód  € VP                  . 4      pVP                  VP                  ) 4      p\        \        V4      V4       VP	                  R VP
                  R7      pVP                  VP                  VP                  V4      4      4      p\        \        V4      V4       VP                  RR.4      pVP                  R\        P                  ! R4      ,           4      p\        \        V4      V4       RpVP                  V3R4      pVP                  R\        P                  ! V4      ,           4      p\        \        V4      V4       VP                  R.R,          4      pVP                  V4      pVP                  Wf.4      pVP                  Ww.4      p	\        VP                  \        V	4      4      VP                  V4      4       \        VP                  \        V	^ R	7      4      VP                  V^ R	7      4       \        VP                  \        V	^R	7      4      VP                  V^R	7      4       VP                  VP                  .4      p
VP                  VP                  .4      p\        \        V
4      V
^ ,          4       \        \        V
) 4      V
^ ,          ) 4       \        \        V4      V^ ,          4       \        \        VP                  VP                  ) VP                  ) .4      4      V
^ ,          ) 4       VP                  R
R.R\        P                  ) ..4      pVP                  R
R.4      p\        \        VRR	7      V4       VP                  VRR	7      p\        \        VRRR7      V4       \        \        VRR	7      VP                  R
4      4       R# )ç      i@©Údtypeç     @�@ç       @éè  é'  g     ˆÃ@çœWw'&l¡7i@B ©Úaxisg    _ Bg»½×Ùß|Û=T)r*   ÚkeepdimsNg    _ Âr   )r   éþÿÿÿ)r   Úinfr   r   ÚarangeÚfloat64ÚlogÚsumÚexpr   r   ÚfullÚstackÚnanÚnpÚexpand_dims)Úselfr   ÚaÚdesiredÚbÚnr   ÚlogxÚXÚlogXr-   r5   r   s   &&           r   Útest_logsumexpÚTestLogSumExp.test_logsumexp%   sÄ  € à�J‰J�r‹NˆØ—*‘*˜bŸf™f˜WÓ%ˆÜœ	 !› gÔ.ð �I‰I�d "§*¡*ˆIÓ-ˆØ—&‘&˜Ÿ™ §¡ q£	Ó*Ó+ˆÜœ	 !› gÔ.ð �J‰J˜˜u�~Ó&ˆØ—*‘*˜V¤d§h¢h¨s£mÕ3Ó4ˆÜœ	 !› gÔ.àˆØ�G‰G�Q�D˜%Ó ˆØ—*‘*˜W¤t§x¢x°£{Õ2Ó3ˆÜœ	 !› gÔ.à�J‰J˜�w Õ(Ó)ˆØ�v‰v�a‹yˆØ�H‰H�a�VÓˆØ�x‰x˜˜Ó%ˆÜ˜Ÿ™œy¨›Ó/°·±¸³Ô;Ü˜Ÿ™œy¨°AÔ6Ó7¸¿¹ÀÈ¸Ó9JÔKÜ˜Ÿ™œy¨°AÔ6Ó7¸¿¹ÀÈ¸Ó9JÔKð �j‰j˜"Ÿ&™&˜Ó"ˆØ�j‰j˜"Ÿ&™&˜Ó"ˆÜœ	 #›¨¨A­Ô/Üœ	 3 $›¨#¨a­&¨Ô1Üœ	 #›¨¨A­Ô/Üœ	 "§*¡*¨r¯v©v¨g¸¿¹°wÐ-?Ó"@ÓAÀCÈÅFÀ7ÔKð �J‰J˜˜u˜Ø¤§¡ Ð(ð*ó +ˆà�j‰j˜$ ˜Ó'ˆÜœ	 !¨"Ô-¨sÔ3ð �n‰n˜S rˆnÓ*ˆÜœ	 !¨"°tÔ<¸cÔBô 	œ	 !¨(Ô3°R·Z±ZÀÓ5EÖFr   c           
     ó  € VP                  R VP                  R7      pVP                  R RR4      pVP                  VP                  W1P	                  V4      ,          4      4      p\        \        W#R7      V4       VP                  RR.4      pVP                  RR.4      pVP                  R\        P                  ! R4      ,           4      p\        \        W#R7      V4       VP                  R.R,          4      pVP                  ^RR4      pVP                  V4      pVP                  WU34      pVP                  Wf34      pVP                  W334      p	\        VP	                  \        W‰R7      4      VP                  W—,          4      4       \        VP	                  \        W‰^ R7      4      VP                  W—,          ^ R	7      4       \        VP	                  \        W‰^R7      4      VP                  W—,          ^R	7      4       R
# )r!   r"   ç        ©r;   r&   g333333ó?r(   i † )r;   r*   r)   Nç      ð¿g333333@)r.   r/   r0   r1   r2   r   r   r   r   Úlinspacer4   )
r8   r   r9   r;   r:   r   r=   r>   r?   ÚBs
   &&        r   Útest_logsumexp_bÚTestLogSumExp.test_logsumexp_bW   sƒ  € Ø�I‰I�d "§*¡*ˆIÓ-ˆØ�I‰I�d˜B Ó$ˆØ—&‘&˜Ÿ™ §&¡&¨£)¥Ó,Ó-ˆÜœ	 !Ô)¨7Ô3à�J‰J˜˜d�|Ó$ˆØ�J‰J˜˜S�zÓ"ˆØ—*‘*˜T¤D§H¢H¨WÓ$5Õ5Ó6ˆÜœ	 !Ô)¨7Ô3à�J‰J˜�w Õ'Ó(ˆØ�K‰K˜˜4 Ó(ˆØ�v‰v�a‹yˆØ�H‰H�a�VÓˆØ�x‰x˜˜Ó%ˆØ�H‰H�a�VÓˆÜ˜Ÿ™œy¨Ô3Ó4°b·f±f¸Q½U³mÔDÜ˜Ÿ™œy¨¸Ô;Ó<¸b¿f¹fÀQÅUÐQR¸fÓ>SÔTÜ˜Ÿ™œy¨¸Ô;Ó<¸b¿f¹fÀQÅUÐQR¸fÓ>SÖTr   c                óØ   € VP                  . RO4      pVP                  . RO4      p\        W#RR7      w  rE\        WAP                  R4      4       \        WQP                  R4      4       R# )r   T©r;   Úreturn_signç      ð?N)r   r   r   )r   r   r   rE   )r   r   r   r   ©r8   r   r9   r;   ÚrÚss   &&    r   Útest_logsumexp_signÚ!TestLogSumExp.test_logsumexp_signl   sL   € Ø�J‰J’yÓ!ˆØ�J‰J’{Ó#ˆä˜¨TÔ2‰ˆÜ˜Ÿ:™: b›>Ô*Ü˜Ÿ:™: c›?Ö+r   c                óø   € VP                  ^^.4      pVP                  ^R.4      p\        W#RR7      w  rEVP                  V4      '       d   Q hVP                  V4      '       d   Q hV^ 8  g   Q hV^ 8X  g   Q hR# )r   TrK   Nr   )r   r   ÚisfiniteÚisnanrN   s   &&    r   Útest_logsumexp_sign_zeroÚ&TestLogSumExp.test_logsumexp_sign_zerot   so   € Ø�J‰J˜˜1�vÓˆØ�J‰J˜˜2�wÓˆä˜¨TÔ2‰ˆØ—;‘;˜q—>’>Ð!Ð!Ø—8‘8˜A—;’;ÐˆØ�1ŒuˆˆuØ�AŒvˆŠvr   c                ó.  € VP                  R4      pVP                  V4      p\        V^VRR7      w  rEVP                  VP                  u;8X  d	   R8X  g   Q h Q h\        VRVRR7      w  rEVP                  VP                  u;8X  d	   R8X  g   Q h Q hR# )r   T)r*   r;   rL   N©r   é   é   é   ©r   rZ   r\   ©r   r[   ©ÚonesÚ	ones_liker   ÚshaperN   s   &&    r   Útest_logsumexp_sign_shapeÚ'TestLogSumExp.test_logsumexp_sign_shape~   sƒ   € Ø�G‰G�LÓ!ˆØ�L‰L˜‹Oˆä˜  a°TÔ:‰ˆØ�w‰w˜!Ÿ'™'Ö. YÔ.Ð.Ñ.Ð.Ð.ä˜ ¨1¸$Ô?‰ˆØ�w‰w˜!Ÿ'™'Ö* UÔ*Ð*Ñ*Ð*Ò*r   c                ó  € VP                  . RO4      p\        VRR7      w  r4VP                  VP                  V4      4      pWQP	                  V4      ,          p\        WF4       \        WAP                  V4      ,          V4       R# )r   T)rL   N)y      ð?      ð?y       @      ð¿y       À      @)r   r   r1   r2   Úabsr   )r8   r   r9   rO   rP   Úexpected_sumexpÚexpected_signs   &&     r   Útest_logsumexp_complex_signÚ)TestLogSumExp.test_logsumexp_complex_signˆ   sb   € Ø�J‰JÒ0Ó1ˆä˜¨Ô-‰ˆàŸ&™& §¡¨£Ó+ˆà'¯&©&°Ó*AÕAˆä˜Ô)Ü˜ŸF™F 1›I� Ö7r   c                óÎ   € VP                  R4      pVP                  V4      p\        V^VR7      pVP                  R8X  g   Q h\        VRVR7      pVP                  R8X  g   Q hR# )r   )r*   r;   NrY   r]   r^   r_   )r8   r   r9   r;   rO   s   &&   r   Útest_logsumexp_shapeÚ"TestLogSumExp.test_logsumexp_shape”   s]   € Ø�G‰G�LÓ!ˆØ�L‰L˜‹Oˆä�a˜a 1Ô%ˆØ�w‰w˜)Ô#Ð#Ð#ä�a˜f¨Ô*ˆØ�w‰w˜&Ô Ð Ò r   c                óœ   € VP                  ^R.4      pVP                  ^^ .4      p\        \        W#R7      VP                  R4      4       R# )r   r'   rD   rM   N)r   r   r   ©r8   r   r9   r;   s   &&  r   Útest_logsumexp_b_zeroÚ#TestLogSumExp.test_logsumexp_b_zerož   s=   € Ø�J‰J˜˜5�zÓ"ˆØ�J‰J˜˜1�vÓˆäœ	 !Ô)¨2¯:©:°b«>Ö:r   c                ób   € VP                  R4      pVP                  R4      p\        W#R7       R# )r\   rD   N)r\   r   rZ   r   )r[   r   é   )Úzerosr`   r   ro   s   &&  r   Útest_logsumexp_b_shapeÚ$TestLogSumExp.test_logsumexp_b_shape¤   s&   € Ø�H‰H�\Ó"ˆØ�G‰G�IÓˆä�!×r   Úargc                óŒ   € \        V4      \        \        P                  ! \        P                  ! V4      4      4      8X  g   Q hR # ©N)r   r6   r   Ú
atleast_1d)r8   rw   s   &&r   Útest_xp_invalid_inputÚ#TestLogSumExp.test_xp_invalid_inputª   s+   € ä˜‹~¤¬2¯:ª:´b·m²mÀCÓ6HÓ+IÓ!JÔJÐJÒJr   c                óš   € R R .p\         P                  ! R\        P                  ! R4      ,           4      p\	        \        V4      V4       R# )r&   r$   r%   N)r6   r   r   r0   r   r   )r8   r9   r:   s   &  r   Útest_array_likeÚTestLogSumExp.test_array_like®   s4   € Ø�4ˆLˆÜ—*’*˜V¤d§h¢h¨s£mÕ3Ó4ˆÜœ	 !› gÖ.r   r#   c                ó6  € \        W!4      pVP                  R R .VR7      pVP                  VR4      '       d   VP                  R4      P                  MTpVP                  R \        P
                  ! R4      ,           VR7      p\        \        V4      V4       R# )r$   r"   ÚintegralrM   r%   N)Úgetattrr   Úisdtyper#   r   r0   r   r   )r8   r#   r   r9   Údesired_dtyper:   s   &&&   r   Útest_dtypes_aÚTestLogSumExp.test_dtypes_a³   sx   € ä˜Ó"ˆØ�J‰J˜˜u�~¨UˆJÓ3ˆØ13·±¸EÀ:×1NÒ1N˜Ÿ™ B›×-Ò-Ø#ð 	à—*‘*˜V¤d§h¢h¨s£mÕ3¸=�*ÓIˆÜœ	 !› gÖ.r   Údtype_aÚdtype_bc                óà  € \        W14      p\        W24      pVP                  ^^.VR7      pVP                  ^R.VR7      p\        V4      '       dx   WE3 Uu. uF  pVP                  VR4      '       d   K  VNK!  	  p	p\	        V	4      ^8  d+   V	P                  VP                  R4      P                  4       VP                  ! V	!  p
M4VP                  WE4      p
VP                  V
R4      '       d   \        V4      p
VP                  \        P                  ! \        P                  ! ^4      \        P                  ! ^4      ,
          4      V
R7      p\        \        WgR7      V4       R# u upi )rZ   r"   r�   rM   rD   Nr   )r‚   r   r   rƒ   ÚlenÚappendr#   Úresult_typer   r   r0   r2   r   r   )r8   r‡   rˆ   r   Ú
xp_dtype_aÚ
xp_dtype_br9   r;   r#   Úxp_float_dtypesr„   r:   s   &&&&        r   Útest_dtypes_abÚTestLogSumExp.test_dtypes_ab¼   s(  € ô ˜RÓ)ˆ
Ü˜RÓ)ˆ
Ø�J‰J˜˜1�v ZˆJÓ0ˆØ�J‰J˜˜2�w jˆJÓ1ˆÜ˜r×"Ò"ð 4>Ñ2Jó EÑ2J¨Ø&(§j¡j°¸
×&C÷  %˜uÑ2JˆOð Eä�?Ó# aÔ'Ø×&Ñ& r§z¡z°"£~×';Ñ';Ô<ØŸNšN¨OÑ<‰MàŸN™N¨:ÓBˆMØ�z‰z˜-¨×4Ò4Ü/°Ó3ˆ}Ø—*‘*œTŸXšX¤d§h¢h¨q£k´D·H²H¸Q³KÕ&?Ó@È�*ÓVˆÜœ	 !Ô)¨7Ö3ùòEs   ÁE+Á6E+c                ó˜   € VP                  R R.4      p\        V4      pVP                  V^ ,          V^,          4      p\        W44       R# )rC   Ng      DÀ)r   r   Ú	logaddexpr   )r8   r   r9   r   r   s   &&   r   Útest_gh18295ÚTestLogSumExp.test_gh18295Ò   s?   € ð �J‰J˜˜U�|Ó$ˆÜ˜‹lˆØ�l‰l˜1˜Q�4  1¥Ó&ˆÜ˜Ö!r   Ú	complex64Ú
complex128c                ón  € \         P                  P                  R 4      p\        W4      pRpVP	                  ^^(V4      RVP	                  ^^(V4      ,          ,           pVP                  WRR7      p\        V^R7      pVP                  VP                  VP                  V4      ^R7      4      pVP                  VP                  V4      VP                  4      p\        VP                  VP                  V4      4      V4       \        Wg4       \        VR^R7      w  ršVP                  VP                  V4      ^R7      p\        VP                  VP                  V
4      4      V4       \        W‘P!                  VP                  V4      4      4       \        W§VP                  V4      ,          4       R# )l   &¦DN$ù              ð?r"   r)   T)rL   r*   N)é
   éd   )r6   ÚrandomÚdefault_rngr‚   Úuniformr   r   r0   r1   r2   Ú	full_likeÚimagr   r	   rf   r   Úreal)r8   r   r#   Úrngrb   r   r   r   ÚmaxÚoutÚsgns   &&&        r   Útest_gh21610ÚTestLogSumExp.test_gh21610Ú   sA  € ô �i‰i×#Ñ#Ð$6Ó7ˆÜ˜Ó"ˆØˆØ�K‰K˜˜2˜uÓ%¨¨c¯k©k¸!¸RÀÓ.GÕ(GÕGˆØ�J‰J�qˆJÓ&ˆä˜ Ô"ˆØ�f‰f�R—V‘V˜BŸF™F 1›I¨A�VÓ.Ó/ˆØ�l‰l˜2Ÿ7™7 3›<¨¯©Ó/ˆÜ�r—v‘v˜bŸg™g c›lÓ+¨SÔ1Ü˜Ô!ä˜Q¨D°qÔ9‰ˆØ�f‰f�R—V‘V˜A“Y QˆfÓ'ˆÜ�r—v‘v˜bŸg™g c›lÓ+¨SÔ1Ü˜ŸW™W R§V¡V¨C£[Ó1Ô2Ü˜ §¡¨£�_Ö-r   c                ó  € VP                  ^ R.4      p\        V4      pVP                  R4      p\        VP                  V4      VP                  V4      4       \        VP	                  V4      VP	                  V4      ^ RR7       R# )r   y        š™™™™™y<gVçž¯Ò<)r   ÚrtolNyï9úþB.æ?š™™™™™i<)r   r   r   r¡   r    )r8   r   r   r   r   s   &&   r   Útest_gh21709_small_imaginaryÚ*TestLogSumExp.test_gh21709_small_imaginaryó   sj   € ð �J‰J˜Ð5Ð6Ó7ˆÜ˜‹lˆð
 �j‰jÐCÓDˆÜ˜Ÿ™ › b§g¡g¨c£lÔ3Ü˜Ÿ™ › b§g¡g¨c£l¸À×Gr   zx,yrC   g%ušè?g}®¶bÙ@gÑ"Ûù~j@g’ËH¿ý@)Úrepeatc           
     óÐ   € \        VP                  W.4      4      pVP                  VP                  VP	                  VP                  W.4      4      4      4      p\        WE4       R # ry   )r   r   r0   r1   r2   r   )r8   r   Úyr   r   r   s   &&&&  r   Útest_gh22601_infinite_elementsÚ,TestLogSumExp.test_gh22601_infinite_elements  sL   € ô4 ˜Ÿ
™
 A 6Ó*Ó+ˆØ�f‰f�R—V‘V˜BŸF™F 2§:¡:¨q¨fÓ#5Ó6Ó7Ó8ˆÜ˜Ö!r   c                óð   € VP                  RR.4      pVP                  RR.4      p\        V4       \        W#R7       \        W!P                  RR.4      4       \        W1P                  RR.4      4       R# )zFTest that logsumexp doesn't accidentally write back to its parameters.g      @g      @g      @r%   rD   N)r   r   r   ro   s   &&  r   Útest_no_writebackÚTestLogSumExp.test_no_writeback  s_   € à�J‰J˜˜B�xÓ ˆØ�J‰J˜˜B�xÓ ˆÜ�!ŒÜ�!ÕÜ˜Ÿ:™: r¨2 hÓ/Ô0Ü˜Ÿ:™: r¨2 hÓ/Ö0r   Úx_rawrM   r™   c                óÎ   € V F^  pVP                  WR7      p\        \        V4      4      \        V4      8X  g   Q h\        \        WUR7      4      \        V4      8X  d   K^  Q h	  R# )z(Test input device propagation to output.)ÚdevicerD   N)r   r   r   )r8   r´   r   ÚdevicesÚdr   s   &&&&  r   Útest_deviceÚTestLogSumExp.test_device(  sU   € ó ˆAØ—
‘
˜5�
Ó+ˆAÜœY q›\Ó*¬i¸«lÔ:Ð:Ð:ÜœY qÔ.Ó/´9¸Q³<Ö?Ð?Ð?ó r   c           	     ó&  € VP                  . RO4      pVP                  . RO4      p\        \        W#R7      \        W#RR7      ^ ,          4       VP                  . RO4      p\        \        W#R7      VP                  VP                  4      4       R# )çr¯£ÜCƒ@rD   TrK   N)r¼   gå� X××?g™	mÖlÿ@)çª_+™Jþ?çãÌeYæ»@g®|H•4ë¿)r½   r¾   g«R%ÍÀ)r   r   r   r5   ro   s   &&  r   Útest_gh22903ÚTestLogSumExp.test_gh229030  sj   € ð �J‰JÒ;Ó<ˆØ�J‰JÒ<Ó=ˆÜœ	 !Ô)¬9°QÈÔ+NÈqÕ+QÔRð �J‰JÒ<Ó=ˆÜœ	 !Ô)¨2¯:©:°b·f±fÓ+=Ö>r   za, b, sign_refNc                óØ   € VP                  V4      Ve   VP                  V4      MR r2\        W#RR7      w  rVVP                  V4      '       g   Q h\        WaP                  V4      4       R # )NTrK   )r   r   Úisinfr   )r8   r   r9   r;   Úsign_refÚvalÚsigns   &&&&&  r   Útest_gh23548ÚTestLogSumExp.test_gh23548>  sR   € ð �z‰z˜!‹}¨qª}˜bŸj™j¨œmÀ$ˆ1Ü˜a°$Ô7‰	ˆØ�x‰x˜�}Š}Ðˆ}Ü˜Ÿj™j¨Ó2Ö3r   © )r   rZ   r[   g       €rE   )$Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r@   rH   rQ   rV   rc   ri   rl   rp   ru   ÚpytestÚmarkÚparametrizer{   r~   Údtypesr…   r�   r”   r¦   rª   ÚitÚproductr6   r-   Úcomplexr¯   r²   r¹   r¿   rÆ   Ú__static_attributes__Ú__classdictcell__©Ú__classdict__s   @r   r   r       sÐ  ø‡ € ò
0GòdUò*,òò+ò
8ò!ò;òð ‡[�[×Ñ˜U Qª	 NÓ3ñKó 4ðKò/ð
 ‡[�[×Ñ˜W fÓ-ñ/ó .ð/ð ‡[�[×Ñ˜Y¨Ó/Ø‡[�[×Ñ˜Y¨Ó/ñ4ó 0ó 0ð4ò("ð ‡[�[×Ñ˜W {°LÐ&AÓBñ.ó Cð.ò0Hð ‡[�[×Ñ˜U B§J¢Jà�V‰VˆGØ�F‰FÙ�R—V‘V�G˜RÓ Ù�R—V‘V�G˜SÓ!Ù�R—V‘V�G˜RŸV™VÓ$Ù�R—V‘V�G˜bŸf™f˜WÓ%Ù�B—F‘F˜BÓÙ�B—F‘F˜CÓ Ù�B—F‘F˜BŸF™FÓ#Ù�B—F‘F˜RŸV™V˜GÓ$á�R—V‘V�G˜VÓ$Ù�R—V‘V�G˜VÓ$Ù�R—V‘V�G˜VÓ$Ù�R—V‘V�G˜VÓ$Ù�B—F‘F˜FÓ#Ù�B—F‘F˜FÓ#Ù�B—F‘F˜FÓ#Ù�B—F‘F˜FÓ#ð'	
ð( ô+%ó ñ."ó/ð."ò1ð ‡[�[×Ñ˜W s¨D°" oÓ6ñ@ó 7ð@ò?ð ‡[�[×ÑÐ-Ø "§¡˜x¨¨rÐ2Ø "§¡˜x¨#¨°Ð4ð6ó7ñ4ó7ö4r   r   c                   ó~   a € ] tR tRt o R tR t]P                  P                  RRR7      R 4       t	R t
R	 tR
tV tR# )ÚTestSoftmaxiJ  c                ó   € \        \        VP                  . RO4      4      VP                  . R	O4      RR7       \        \        VP                  RR.4      4      VP                  RR.4      RR7       \        \        VP                  RR.4      4      VP                  R\        P                  .4      ^\        P                  ,           ,          RR7       VP                  ^VP                  R7      pVP                  . R
OVP                  R7      p\        \        V4      VRR7       \        \        V^d,           4      VRR7       \        \        VP                  VR4      4      VP                  VR4      RR7       R# )r$   rC   rM   ç‚vIhÂ%<=©r©   ç      à?r"   N)r$   rC   rC   rC   ©rM   rC   rC   rC   )g*lI9üi ?g™{O¶?gW-RÎ?gI³ôòšä?©rZ   rZ   )r   r   r   r6   Úer.   r/   Úreshape)r8   r   r   Úexpecteds   &&  r   Útest_softmax_fixturesÚ!TestSoftmax.test_softmax_fixturesL  s)  € Üœ §
¡
Ò+>Ó ?Ó@ØŸ
™
Ò#3Ó4¸5õ	Bäœ §
¡
¨B°¨8Ó 4Ó5ØŸ
™
 B¨ 8Ó,°5õ	:äœ §
¡
¨B°¨8Ó 4Ó5ØŸ
™
 B¬¯© :Ó.°´B·D±DµÕ9Ø"õ	$ð �I‰I�a˜rŸz™zˆIÓ*ˆØ—:‘:ò 3ð ;=¿*¹*ð ó Fˆô
 	œ ›
 H°5Õ9ô 	œ  C¥Ó(¨(¸Õ?ô 	œ §
¡
¨1¨fÓ 5Ó6ØŸ
™
 8¨VÓ4¸5÷	Br   c           	     óp  € \        \        VP                  R R.R R..4      ^ R7      VP                  RR.RR..4      RR7       \        \        VP                  R R.R R..4      ^R7      VP                  RR.RR..4      RR7       VP                  . RO. R	O.4      pVP                  . R
O. RO.4      p\        \        V^R7      VRR7       \        \        VP                  ^ R7      VP                  RR7       VP	                  VR4      p\        \        VRR7      VP	                  VR4      RR7       R# )r$   rC   r)   rÝ   rÛ   rÜ   rM   N)g      9ÀrC   g      9@g      I@)rM   g     Pt@g     h‡@g     p‡@)g£×Ù	Ñ+9g.K|T}%m;gÕð]¤7Š®=g\þÿÿÿï?)rC   g‘pk´&•g†ºTV6Ñ?g¼¢ÕõÔdç?©rZ   rZ   rZ   ©r   rZ   )r   r   r   ÚTrá   )r8   r   r   râ   Úx3ds   &&   r   Útest_softmax_multi_axesÚ#TestSoftmax.test_softmax_multi_axesh  s$  € Üœ §
¡
¨U°B¨K¸%À¸Ð+EÓ FÈQÔOØŸ
™
 R¨ H¨r°2¨hÐ#7Ó8¸uõ	Fäœ §
¡
¨U°B¨K¸%À¸Ð+EÓ FÈQÔOØŸ
™
 R¨ H¨r°2¨hÐ#7Ó8¸uõ	Fð
 �J‰JÒ0Ú0ð2ó 3ˆà—:‘:ò  3ò 4ð	5ó 6ˆô 	œ ¨Ô*¨H¸5ÕAÜœ §¡¨!Ô,¨h¯j©j¸uÕEð �j‰j˜˜IÓ&ˆÜœ ¨&Ô1ØŸ
™
 8¨YÓ7¸e÷	Er   Úarray_api_strictúint->float promotion©Úreasonc                óx   € \        \        VP                  . RO4      4      VP                  . RO4      RR7       R# ©r&   rÛ   rÜ   N)r&   r   r   r   rÞ   )r   r   r   ©r8   r   s   &&r   Útest_softmax_int_arrayÚ"TestSoftmax.test_softmax_int_array‚  s*   € äœ §
¡
ª?Ó ;Ó<ØŸ
™
Ò#3Ó4¸5÷	Br   c                ó\   € \        \        R 4      \        P                  ! R4      RR7       R# )r&   rM   rÛ   rÜ   N©r   r   r6   r   ©r8   s   &r   Útest_softmax_scalarÚTestSoftmax.test_softmax_scalar‡  s   € Üœ ›¤r§z¢z°"£~¸E×Br   c                ód   € \        \        . RO4      \        P                  ! . RO4      RR7       R# rñ   rö   r÷   s   &r   Útest_softmax_array_likeÚ#TestSoftmax.test_softmax_array_likeŠ  s!   € Üœ¢Ó0ÜŸ
š
Ò#3Ó4¸5÷	Br   rÈ   N)rÉ   rÊ   rË   rÌ   rã   rê   rÍ   rÎ   Úxfail_xp_backendsró   rø   rû   rÔ   rÕ   rÖ   s   @r   rÙ   rÙ   J  sP   ø‡ € òBò8Eð4 ‡[�[×"Ñ"Ð#5Ð>TÐ"ÓUñBó VðBòC÷Bð Br   rÙ   c                   ó¢  a € ] tR tRt o R t]P                  P                  RRR7      R 4       tR t	R t
]R	 4       t]R
 4       t]P                  P                  R^ ^d.4      R 4       tR t]P                  P                  R^ ]P$                  ! R4      ]P&                  ! R4      ,          3^RR.RR..3.4      R 4       tR tR tR tRtV tR# )ÚTestLogSoftmaxi�  c                óx   € \        \        VP                  R R.4      4      VP                  RR.4      RR7       R# )r$   rM   rC   rÛ   rÜ   Nç     8�À©r   r
   r   rò   s   &&r   Útest_log_softmax_basicÚ%TestLogSoftmax.test_log_softmax_basic‘  s0   € Üœ B§J¡J°°r¨{Ó$;Ó<ØŸ
™
 B¨ ;Ó/°e÷	=r   rì   rí   rî   c                óx   € \        \        VP                  R ^.4      4      VP                  RR.4      RR7       R# ©r&   rC   rÛ   rÜ   Nr  r  rò   s   &&r   Útest_log_softmax_int_arrayÚ)TestLogSoftmax.test_log_softmax_int_array•  s0   € äœ B§J¡J°°a¨yÓ$9Ó:ØŸ
™
 B¨ ;Ó/°e÷	=r   c                ó4   € \        \        R 4      RRR7       R# )rM   rC   rÛ   rÜ   N)r   r
   r÷   s   &r   Útest_log_softmax_scalarÚ&TestLogSoftmax.test_log_softmax_scalarš  s   € Üœ CÓ(¨#°E×:r   c                ód   € \        \        R ^.4      \        P                  ! RR.4      RR7       R# r  )r   r
   r6   r   r÷   s   &r   Útest_log_softmax_array_likeÚ*TestLogSoftmax.test_log_softmax_array_like�  s'   € Üœ T¨1 IÓ.ÜŸ
š
 B¨ ;Ó/°e÷	=r   c                ó~   € V P                  ^V P                  R7      p. ROpWP                  W P                  R7      3# )r\   r"   ©gùÙ:-‚…ÀgùÙ:-‚…Àgò³uZ÷¿gÉÏÖi,Ü¿)r.   r/   r   ©r   r   Úexpects   &  r   Údata_1dÚTestLogSoftmax.data_1d¡  s9   € à�I‰I�a˜rŸz™zˆIÓ*ˆò(ˆð —*‘*˜V¯:©:�*Ó6Ð6Ð6r   c                ó¦   € V P                  V P                  ^V P                  R7      R4      p. RO. RO.pWP                  W P                  R7      3# )é   r"   )rZ   r\   r  )rá   r.   r/   r   r  s   &  r   Údata_2dÚTestLogSoftmax.data_2d«  sN   € à�J‰J�r—y‘y ¨"¯*©*�yÓ5°vÓ>ˆò)ò)ð	*ˆð —*‘*˜V¯:©:�*Ó6Ð6Ð6r   Úoffsetc                ój   € V P                  V4      w  r4W1,          p\        \        V4      VR R7       R# )rÛ   rÜ   N)r  r   r
   )r8   r  r   r   r  s   &&&  r   Útest_log_softmax_translationÚ+TestLogSoftmax.test_log_softmax_translationº  s+   € ð —L‘L Ó$‰	ˆØ	�ˆÜœ A›¨°U×;r   c                ó¢   € V P                  V4      w  r#VP                  VR4      pVP                  VR4      p\        \        V4      VRR7       R# )rZ   rÛ   rÜ   Nrß   )r  rá   r   r
   ©r8   r   r   r  s   &&  r   Útest_log_softmax_noneaxisÚ(TestLogSoftmax.test_log_softmax_noneaxisÂ  sD   € ð —L‘L Ó$‰	ˆØ�J‰J�q˜&Ó!ˆØ—‘˜F FÓ+ˆÜœ A›¨°U×;r   zaxis_2d, expected_2drÝ   rC   c                ó˜   € VP                  R R.R R..4      p\        \        WAR7      VP                  W$P                  R7      RR7       R# )r$   rM   r)   r"   rÛ   rÜ   N)r   r   r
   r#   )r8   Úaxis_2dÚexpected_2dr   r   s   &&&& r   Ú	test_axesÚTestLogSoftmax.test_axesÊ  sC   € ð
 �J‰J˜ ˜ e¨R [Ð1Ó2ˆÜœ AÔ4ØŸ
™
 ;·g±g˜
Ó>ÀU÷	Lr   c                ó^   € V P                  V4      w  r#\        \        V^R7      VRR7       R# )r   r)   rÛ   rÜ   N)r  r   r
   r  s   &&  r   Útest_log_softmax_2d_axis1Ú(TestLogSoftmax.test_log_softmax_2d_axis1Ó  s%   € Ø—L‘L Ó$‰	ˆÜœ A¨AÔ.°¸U×Cr   c                ó†   € V P                  V4      w  r#\        \        VP                  ^ R7      VP                  RR7       R# )r   r)   rÛ   rÜ   N)r  r   r
   rè   r  s   &&  r   Útest_log_softmax_2d_axis0Ú(TestLogSoftmax.test_log_softmax_2d_axis0×  s-   € Ø—L‘L Ó$‰	ˆÜœ A§C¡C¨aÔ0°&·(±(À×Gr   c                ó¦   € V P                  V4      w  r#VP                  VR4      pVP                  VR4      p\        \        VRR7      VRR7       R# )rZ   r)   rÛ   rÜ   Nræ   rç   )r  rá   r   r
   r  s   &&  r   Útest_log_softmax_3dÚ"TestLogSoftmax.test_log_softmax_3dÛ  sE   € à—L‘L Ó$‰	ˆØ�J‰J�q˜)Ó$ˆØ—‘˜F IÓ.ˆÜœ A¨FÔ3°VÀ%×Hr   rÈ   Nrß   r  )rÉ   rÊ   rË   rÌ   r  rÍ   rÎ   rý   r  r
  r  Ústaticmethodr  r  rÏ   r  r  r6   r0   r`   r$  r'  r*  r-  rÔ   rÕ   rÖ   s   @r   rÿ   rÿ   �  s  ø‡ € ò=ð ‡[�[×"Ñ"Ð#5Ð>TÐ"ÓUñ=ó Vð=ò;ò=ð ñ7ó ð7ð ñ7ó ð7ð ‡[�[×Ñ˜X¨¨3 xÓ0ñ<ó 1ð<ò<ð ‡[�[×ÑÐ3Ø	
ˆB�FŠF�3‹K˜"Ÿ'š' &›/Õ)Ð*Ø	
ˆb�%ˆ[˜2˜u˜+Ð&Ð'ð6ó ñLó	ðLò
DòH÷Ið Ir   rÿ   )Úfloat32r/   r   r   r–   r—   )Ú	itertoolsrÑ   r   rÍ   Únumpyr6   Úscipy._lib._array_apir   r   r   r   Úscipy._lib._array_api_no_0dr   r   r	   Úscipy.specialr
   r   r   Úscipy.special._logsumexpr   rÐ   Úintegral_dtypesr   rÎ   Úfilterwarningsr   rÙ   rÿ   rÈ   r   r   Ú<module>r9     sÿ   ðÛ Û Û ã ÷@ó @÷9ñ 9÷ :Ñ 9Ý 2ò 
M€Ø˜GÐ$€ò&ð ‡�×ÑÐMÓNØ‡�×ÑÐNÓOØ‡�×ÑÐHÓIÙ�9Ó÷c4ð c4ó ó Jó Pó Oðc4ñL	 �7Ó÷ABð ABó ðABñH �;Ó÷PIð PIó  òPIr   