+
    ‹°³jiC  ã                  ó~  € ^ RI Ht ^RIHtHt ^RIHt ^RIHt ]R 4       t]R 4       t	]! ]P                  4      t]P                  ]R 4       4       t]P                  ]]P                  ! R4      R	 4       4       4       t]P                  ]]P                  ! R
4      RPR l4       4       4       t]P                  ]RQR l4       4       t]R 4       t]R 4       t]R 4       t]R 4       t]R 4       t]R 4       t]R 4       t]P                  ]]P2                  ! RRRR7      RRR l4       4       4       t]P                  ]]P2                  ! RRR7      RSR l4       4       4       t]R 4       t]R 4       t]R  4       t]R! 4       t]P                  ]]P2                  ! R"RRR7      RRR# l4       4       4       t ]P                  ]]P2                  ! R$RR7      RSR% l4       4       4       t!]R& 4       t"]R' 4       t#]P                  ]]P2                  ! R(R)R*7      RTR+ R, ll4       4       4       t$]R- 4       t%]P                  ]]P2                  ! R.4      RUR/ l4       4       4       t&]R0 4       t']P                  ]]P2                  ! R14      RQR2 l4       4       4       t(]P                  ]]PR                  ! R3R)R*7      RVR4 R5 ll4       4       4       t*]R6 4       t+]P                  ]]PR                  ! R74      RWR8 l4       4       4       t,]R9 R: l4       t-]R; R< l4       t.]R= R> l4       t/]R? R@ l4       t0]RR]Pb                  3RA RB ll4       t2]R]Pb                  3RC RD ll4       t3]RXRE RF ll4       t4]R]Pb                  3RG RH ll4       t5]RI 4       t6]P                  ]RYRJ l4       4       t7]RK 4       t8]RL RM l4       t9]RN RO l4       t:R# )Zé    )Úannotations)ÚjitÚconstexpr_function)Úcore)Úmathc                óD   € ^ pT pV^8”  d   V^,          pV^,          pK  V# ©r   © )ÚiÚlog2Úns   &  ÚY/home/luca/screen-checking/.venv/lib/python3.14/site-packages/triton/language/standard.pyÚ_log2r   
   s*   € à€DØ	€AØ
ˆaŒ%Ø	ˆa�ˆØ��	ŠØ€Kó    c                ó@   € W ^,
          ,          ^ 8H  ;'       d    V ^ 8g  # ©é   r
   )r   s   &r   Ú_is_power_of_twor      s   € à�Q•�K˜AÑ×(Ð( ! q¡&Ð(r   c                ó.   € W^,
          ,           V,          # )z�
Computes the ceiling division of :code:`x` by :code:`div`

:param x: the input number
:type x: Block
:param div: the divisor
:type div: Block
r
   )ÚxÚdivs   &&r   Úcdivr       s   € ð �q•�M˜cÕ!Ð!r   Úsigmoidc                óL   € ^^\         P                  ! V ) 4      ,           ,          # r   )r   Úexp)r   s   &r   r   r   .   s   € ð �”D—H’H˜a˜R“LÕ Õ!Ð!r   ÚsoftmaxNc                ó°   € Vf   ^ pMTpV \        WVR7      ,
          p\        P                  ! V4      p\        WdVR7      p\        P                  ! WgV4      # ©N©Ú	keep_dims)Úmaxr   r   ÚsumÚfdiv)r   Údimr    Úieee_roundingÚ_dimÚzÚnumÚdens   &&&&    r   r   r   5   sN   € ð ‚{Ø ‰à"ˆØ	ŒC� 9Ô-Õ-€AÜ
�(Š(�1‹+€CÜ
ˆc 9Ô
-€CÜ�9Š9�S˜}Ó-Ð-r   c                óH   € \         P                  ! W P                  .VR7      # )z^
Returns a contiguous flattened view of :code:`x`.

:param x: the input tensor
:type x: Block
)Úcan_reorder)r   ÚreshapeÚnumel)r   r+   s   &&r   Úravelr.   C   s   € ô �<Š<˜ŸG™G˜9°+Ô>Ð>r   c                óÎ   € W,          V,           pWC,          pWV,          pWt,          p\         P                  ! W(,
          V4      pWV,          pW…V,          ,           p	WT,          p
Wš3# )a¥  
Transforms the indices of a row-major `size_i * size_j` matrix into
the indices of a column-major matrix for each group of `size_g` rows.

For example, for :code:`size_i = size_j = 4` and :code:`size_g = 2`, it will
transform ::

    [[0 , 1 , 2 , 3 ],
     [4 , 5 , 6 , 7 ],
     [8 , 9 , 10, 11],
     [12, 13, 14, 15]]

into ::

    [[0, 2,  4 , 6 ],
     [1, 3,  5 , 7 ],
     [8, 10, 12, 14],
     [9, 11, 13, 15]]
©r   Úminimum)r   ÚjÚsize_iÚsize_jÚsize_gÚijÚsize_gjÚgroup_idÚoff_iÚnew_iÚnew_js   &&&&&      r   Ú	swizzle2dr<   O   s[   € ð, 
��a�€Bð �o€Gà�}€HàÕ€Eä�\Š\˜&�.¨&Ó1€Fà	�€Bà˜•KÕ€EØ�L€EØˆ<Ðr   c                ó2   € \         P                  ! V ^ V4      # )a  
Returns a tensor filled with the scalar value 0 for the given :code:`shape` and :code:`dtype`.

:param shape: Shape of the new array, e.g., (8, 16) or (8, )
:type shape: tuple of ints
:param dtype: Data-type of the new array, e.g., :code:`tl.float16`
:type dtype: DType
)r   Úfull)ÚshapeÚdtypes   &&r   ÚzerosrA   w   s   € ô �9Š9�U˜A˜uÓ%Ð%r   c                óB   € \        V P                  V P                  4      # )z{
Returns a tensor of zeros with the same shape and type as a given tensor.

:param input: input tensor
:type input: Tensor
)rA   r?   r@   )Úinputs   &r   Ú
zeros_likerD   „   s   € ô �—‘˜eŸk™kÓ*Ð*r   c                ó¶   € V'       d   W8H  ;'       d    W8  pMR pW8„  ;'       g    Tp\         P                  ! W`V4      p\         P                  ! WaV4      pWx3# ©F©r   Úwhere)	Úvalue1Úindex1Úvalue2Úindex2Útie_break_leftÚtieÚgtÚv_retÚi_rets	   &&&&&    r   Ú_argmax_combinerR   ’   sS   € çØÑ×2Ð2 6¡?‰àˆØ	‰×	Ð	˜C€BÜ�JŠJ�r 6Ó*€EÜ�JŠJ�r 6Ó*€EØˆ<Ðr   c                ó   € \        WW#R 4      # ©T©rR   ©rI   rJ   rK   rL   s   &&&&r   Ú_argmax_combine_tie_break_leftrW   ž   ó   € ä˜6¨6¸4Ó@Ð@r   c                ó   € \        WW#R 4      # rF   rU   rV   s   &&&&r   Ú_argmax_combine_tie_break_fastrZ   £   ó   € ä˜6¨6¸5ÓAÐAr   c                ó.   € \         P                  ! W4      # ©N)r   Úmaximum©ÚaÚbs   &&r   Ú_elementwise_maxrb   ¨   ó   € ä�<Š<˜ÓÐr   r^   Úreturn_indicesÚreturn_indices_tie_break_left)Úreturn_indices_argÚtie_break_argc                ó¶  € \         P                  ! V 4      p V'       dC   V'       d   \         P                  ! W\        VR 7      # \         P                  ! W\        VR 7      # \         P
                  ! V P                  P                  4      \         P
                  ! ^ 4      8  d›   \         P
                  ! V P                  P                  4       4      '       d!   V P                  \         P                  4      p MFV P                  P                  4       '       g   Q R4       hV P                  \         P                  4      p \         P                  ! W\        VR 7      # ©r   z"Expecting input to be integer type)r   Ú_promote_bfloat16_to_float32Ú_reduce_with_indicesrW   rZ   Ú	constexprr@   Úprimitive_bitwidthÚis_floatingÚtoÚfloat32Úis_intÚint32Úreducerb   ©rC   Úaxisrd   re   r    s   &&&&&r   r!   r!   ­   sß   € ô
 ×-Ò-¨eÓ4€Eßß(Ü×,Ò,¨UÔ:XÐdmÔnÐnä×,Ò,¨UÔ:XÐdmÔnÐnä�>Š>˜%Ÿ+™+×8Ñ8Ó9¼D¿NºNÈ2Ó<NÔNÜ�~Š~˜eŸk™k×5Ñ5Ó7×8Ò8ØŸ™¤§¡Ó.‘à—{‘{×)Ñ)×+Ò+ÐQÐ-QÓQÐ+ØŸ™¤§¡Ó,�Ü�{Š{˜5Ô(8ÀIÔNÐNr   zmaximum indexrM   )rg   c                ó&   € \        WR W#R7      w  rEV# ©T)rd   re   r    )r!   ©rC   ru   rM   r    Ú_Úrets   &&&&  r   Úargmaxr{   Â   s   € ô �5¨tÐSaÔw�H€QØ€Jr   c                ó¶   € V'       d   W8H  ;'       d    W8  pMR pW8  ;'       g    Tp\         P                  ! W`V4      p\         P                  ! WaV4      pWx3# rF   rG   )	rI   rJ   rK   rL   rM   rN   ÚltÚ	value_retÚ	index_rets	   &&&&&    r   Ú_argmin_combiner€   Í   sT   € çØÑ×2Ð2 6¡?‰àˆØ	‰×	Ð	˜C€BÜ—
’
˜2 vÓ.€IÜ—
’
˜2 vÓ.€IØÐÐr   c                ó   € \        WW#R 4      # rT   ©r€   rV   s   &&&&r   Ú_argmin_combine_tie_break_leftrƒ   Ù   rX   r   c                ó   € \        WW#R 4      # rF   r‚   rV   s   &&&&r   Ú_argmin_combine_tie_break_fastr…   Þ   r[   r   c                ó.   € \         P                  ! W4      # r]   r0   r_   s   &&r   Ú_elementwise_minr‡   ã   rc   r   r1   c                óŽ  € \         P                  ! V 4      p V'       dC   V'       d   \         P                  ! W\        VR 7      # \         P                  ! W\        VR 7      # \         P
                  ! V P                  P                  4      ^ 8  d›   \         P
                  ! V P                  P                  4       4      '       d!   V P                  \         P                  4      p MFV P                  P                  4       '       g   Q R4       hV P                  \         P                  4      p \         P                  ! W\        VR 7      # ri   )r   rj   rk   rƒ   r…   rl   r@   rm   rn   ro   rp   rq   rr   rs   r‡   rt   s   &&&&&r   Úminr‰   è   sÖ   € ô
 ×-Ò-¨eÓ4€Eßß(Ü×,Ò,¨UÔ:XÐdmÔnÐnä×,Ò,¨UÔ:XÐdmÔnÐnä�>Š>˜%Ÿ+™+×8Ñ8Ó9¸BÔ>Ü�~Š~˜eŸk™k×5Ñ5Ó7×8Ò8ØŸ™¤§¡Ó.‘à—{‘{×)Ñ)×+Ò+ÐQÐ-QÓQÐ+ØŸ™¤§¡Ó,�Ü�{Š{˜5Ô(8ÀIÔNÐNr   zminimum indexc                ó&   € \        WR W#R7      w  rEV# rw   )r‰   rx   s   &&&&  r   Úargminr‹   ý   s   € ô �¨TÐQ_Ôu�F€AØ€Jr   c                ó   € W,           # r]   r
   r_   s   &&r   Ú_sum_combiner�     ó	   € à�5€Lr   c                ó  € Ve   V# R pV P                  4       '       d(   V P                  ^ 8  d   \        P                  pV# R pV# V P	                  4       '       d$   V P                  ^ 8  d   \        P
                  MR pV# r]   )Úis_int_signedÚint_bitwidthr   rr   Úis_int_unsignedÚuint32)Úin_dtyper@   Ú	out_dtypes   && r   Ú_pick_sum_dtyper–     s�   € àÒØˆð €IØ×Ñ×ÒØ"*×"7Ñ"7¸"Ô"<”D—J‘Jˆ	ð Ðð CGˆ	ð Ðð 
×	!Ñ	!×	#Ò	#Ø#+×#8Ñ#8¸2Ô#=”D—K’KÀ4ˆ	ØÐr   r"   r@   )Ú	dtype_argc               ó   € V ^8„  d   QhRR/# ©é   r@   úcore.constexprr
   )Úformats   "r   Ú__annotate__r�     s   € ÷ Gñ G°.ñ Gr   c                ó’   € \        V P                  V4      pVe   V P                  V4      p \        P                  ! W\
        VR7      # r   )r–   r@   ro   r   rs   r�   )rC   ru   r    r@   r•   s   &&&& r   r"   r"     s;   € ô
 !0°·±¸UÓ C€IàÒØ—‘˜Ó#ˆÜ�;Š;�u¤L¸IÔFÐFr   c                ó   € W,          # r]   r
   r_   s   &&r   Ú_xor_combiner    (  rŽ   r   zxor sumc                ó®   € \         P                  ! V P                  P                  P	                  4       R 4       \         P
                  ! W\        VR7      # )z#xor_sum only supported for integersr   )r   Ústatic_assertÚtypeÚscalarrq   rs   r    ©rC   ru   r    s   &&&r   Úxor_sumr¦   0  s;   € ô 	×Ò�u—z‘z×(Ñ(×/Ñ/Ó1Ð3XÔYÜ�;Š;�u¤L¸IÔFÐFr   c                ó   € W,          # r]   r
   )r   Úys   &&r   Ú_or_combiner©   ;  rŽ   r   Ú	reduce_orc                ó®   € \         P                  ! V P                  P                  P	                  4       R 4       \         P
                  ! W\        VR7      # )z%reduce_or only supported for integersr   )r   r¢   r£   r¤   rq   rs   r©   r¥   s   &&&r   rª   rª   @  s;   € ô 	×Ò�u—z‘z×(Ñ(×/Ñ/Ó1Ð3ZÔ[Ü�;Š;�u¤K¸9ÔEÐEr   Úcumsumc               ó   € V ^8„  d   QhRR/# r™   r
   )rœ   s   "r   r�   r�   N  s   € ÷ 	Eñ 	E°ñ 	Er   c                ó¼   € \         P                  ! V 4      p \        V P                  V4      pVe   V P	                  V4      p \         P
                  ! W\        V4      # r]   )r   rj   r–   r@   ro   Úassociative_scanr�   )rC   ru   Úreverser@   r•   s   &&&& r   r¬   r¬   K  sM   € ô ×-Ò-¨eÓ4€EÜ /°·±¸UÓ C€IàÒØ—‘˜Ó#ˆä× Ò  ¬l¸GÓDÐDr   c                ó   € W,          # r]   r
   r_   s   &&r   Ú_prod_combiner²   ]  rŽ   r   Úcumprodc                óf   € \         P                  ! V 4      p \         P                  ! W\        V4      # r]   )r   rj   r¯   r²   )rC   ru   r°   s   &&&r   r³   r³   b  s)   € ô
 ×-Ò-¨eÓ4€EÜ× Ò  ¬m¸WÓEÐEr   c               ó    € V ^8„  d   QhRRRR/# )rš   Ún_dimsr›   r2   r
   )rœ   s   "r   r�   r�   o  s   € ÷ ñ �~ð ¨.ñ r   c                óº   € \         P                  ! ^ ^4      p\         P                  ! V^.W,
          ^,
          ,          ^.,           ^.V,          ,           4      pV# r	   )r   Úaranger,   )r¶   r2   Úars   && r   Ú
_indicatorrº   n  sF   € ä	�Š�Q˜Ó	€BÜ	�Š�b˜1˜# ¥¨a¥Õ0°A°3Õ6¸!¸¸q½Õ@Ó	A€BØ€Ir   c               ó   € V ^8„  d   QhRR/# )rš   r   r›   r
   )rœ   s   "r   r�   r�   v  s   € ÷ ñ  .ñ r   c                ón  € \        V P                  4      p\        V P                  P                  R R7      pV P                  VR R7      pV\        WS^,
          V,
          R 4      ,          pVP                  V P                  R R7      p\        W24      p\        P                  ! W8„  W,          8g  Wp4      p	V	# )T©ÚbitwidthÚsigned©Úbitcast)
r   r-   Ú_get_int_dtyper@   rm   ro   r¦   rº   r   rH   )
r   Úflipr   r¶   ÚidtypeÚixÚiyr¨   Úis_rightrz   s
   &&&       r   Ú_compare_and_swaprÈ   u  s–   € ô # 1§7¡7›^€Fô  Q§W¡W×%?Ñ%?ÈÔM€FØ	
�‰ˆf˜dˆÓ	#€BØ	Œg�b 1�* q�.¨$Ó/Õ	/€BØ
�‰ˆa�g‰g˜tˆÓ$€Aô ˜&Ó$€Hô �*Š*�a‘e ¥Ñ1°1Ó
8€CØ€Jr   c               ó    € V ^8„  d   QhRRRR/# )rš   Ústager›   Úorderr
   )rœ   s   "r   r�   r�   ‰  s   € ÷ ñ  ~ð ¸nñ r   c                óÆ   € V^8X  d!   \        \        V P                  4      V4      pMTp\        P                  ! V4       F  p\        WV^,
          V,
          4      p K  	  V # )zR
order_type 0 == ascending
order_type 1 == descending
order_type 2 == alternating
)rº   r   r-   r   Ústatic_rangerÈ   )r   rÊ   rË   rÃ   r   s   &&&  r   Ú_bitonic_merge_hypercuberÎ   ˆ  sR   € ð �„zÜœ% §¡›.¨%Ó0‰àˆä×Ò˜uÖ%ˆÜ˜a u¨q¥y°1¥}Ó5Šñ &à€Hr   c               ó$   € V ^8„  d   QhRRRRRR/# )rš   rÊ   r›   rË   r¶   r
   )rœ   s   "r   r�   r�   Ÿ  s!   € ÷ ñ ˜^ð °Nð ÈNñ r   c                óÂ   € \         P                  ! V ^.\        V P                  4      ,          4      p\	        WAV4      p\         P                  ! W@P
                  4      p V # )rš   )r   r,   r   r-   rÎ   r?   )r   rÊ   rË   r¶   Úhs   &&&& r   Ú_bitonic_mergerÒ   ž  sD   € ä�Š�Q˜˜œe A§G¡G›nÕ,Ó-€AÜ  ¨5Ó1€AÜ�Š�QŸ™Ó €AØ€Hr   c               ó$   € V ^8„  d   QhRRRRRR/# ©rš   Úkr›   r$   Ú
descendingr
   )rœ   s   "r   r�   r�   §  s"   € ÷ %ñ %�Nð %°ð %ÐSañ %r   c                ó¦  € Vf   \        V P                  4      ^,
          MTp\        P                  ! V\        V P                  4      ^,
          8H  R4       \	        V P                  V,          4      pVf   TM
\	        V4      p\	        V P
                  4      p\        P                  ! Y'       d   ^.V,          M^.4      p\        P                  ! ^V^,           4       F  p	\        Y‰W•8  d   ^MT4      pK  	  \        P                  ! V^,           V^,           4       F{  p	V'       d/   \        V\	        VP
                  4      ^,
          V,
          R7      M-\        V\	        VP
                  4      ^,
          V,
          R7      p\        Y†W•8  d   ^MT4      pK}  	  \        P                  ! W€P                  RR ^V,          .,           4      p V # )aA  
Sorts a tensor along a specified dimension.

:param x: The input tensor to be sorted.
:type x: Tensor
:param dim: The dimension along which to sort the tensor. If None, the tensor is sorted along the last dimension. Currently, only sorting along the last dimension is supported.
:type dim: int, optional
:param k: the number of top elements to select. If none, assume k = x.shape[dim]
:type k: int, optional
:param descending: If set to True, the tensor is sorted in descending order. If set to False, the tensor is sorted in ascending order.
:type descending: bool, optional
Nú+only minor dimension is currently supported)ru   éÿÿÿÿ)Úlenr?   r   r¢   r   r-   r,   rÍ   rÎ   r!   r‰   )
r   rÕ   r$   rÖ   r&   Úlog_nÚlog_kr¶   rÑ   r   s
   &&&&      r   Ú	sort_implrÝ   ¦  sY  € ð 03ª{œ3˜qŸw™w›<¨!Ö+À€DÜ×Ò�tœs 1§7¡7›|¨aÕ/Ñ/Ð1^Ô_ä! !§'¡'¨$¥-Ó0€EØ%&¢Y™E´E¸!³H€Eä" 1§7¡7›^€Fô 	�Š�Q®˜˜˜fž°Q°CÓ8€Aô ×Ò˜q %¨!¥)Ö,ˆÜ$ Q°´	©1¸zÓJŠñ -ô
 ×Ò˜u q�y¨%°!­)Ö4ˆß9CŒC�œ˜qŸw™w›¨!Õ+¨eÕ3Õ5ÌÈQÔV[Ð\]×\cÑ\cÓVdÐghÕVhÐkpÕVpÔIrˆÜ$ Q°A´I©qÀ:ÓNŠñ 5ô
 	�Š�QŸ™  ˜¨¨5­ zÕ1Ó2€AØ€Hr   c               ó    € V ^8„  d   QhRRRR/# ©rš   r$   r›   rÖ   r
   )rœ   s   "r   r�   r�   Ð  s   € ÷ 8ñ 8�ð 8°Nñ 8r   c                ó   € \        WVR 7      # ))r$   rÖ   ©rÝ   )r   r$   rÖ   s   &&&r   Úsortrâ   Ï  s   € ä�Q¨JÔ7Ð7r   c               ó$   € V ^8„  d   QhRRRRRR/# rÔ   r
   )rœ   s   "r   r�   r�   Õ  s!   € ÷ =ñ =ˆ~ð = Nð =À~ñ =r   c                ó   € \        WW#R7      # )aL  
Returns the k largest (or smallest) elements of the input tensor along the specified dimension.

The elements are returned in sorted order (largest first).

:param x: The input tensor.
:type x: Tensor
:param k: The number of top elements to return. Must be a power of two.
:type k: int
:param dim: The dimension along which to find the top k elements.
            If None, uses the last dimension. Currently only the last dimension is supported.
:type dim: int, optional
:param descending: If set to True, returns k largest elements. If set to False, returns k smallest elements.
:type descending: bool, optional
:return: A tensor containing the k largest elements along the specified dimension.
:rtype: Tensor

Example::

    # Get top 4 elements from a 1D tensor
    x = tl.arange(0, 16)
    top4 = tl.topk(x, 4)  # Returns [15, 14, 13, 12]
)rÕ   r$   rÖ   rá   )r   rÕ   r$   rÖ   s   &&&&r   Útopkrå   Ô  s   € ô2 �Q Ô<Ð<r   c               ó    € V ^8„  d   QhRRRR/# rß   r
   )rœ   s   "r   r�   r�   ñ  s   € ÷ 9ñ 9˜.ð 9¸^ñ 9r   c                óþ   € Vf   \        V P                  4      ^,
          MTp\        P                  ! V\        V P                  4      ^,
          8H  R4       \	        V P                  R,          4      p\        WW$4      # )NrØ   rÙ   )rÚ   r?   r   r¢   r   rÒ   )r   r$   rÖ   r&   r¶   s   &&&  r   Úbitonic_mergerè   ð  s^   € ð 03ª{œ3˜qŸw™w›<¨!Ö+À€DÜ×Ò�tœs 1§7¡7›|¨aÕ/Ñ/Ð1^Ô_Ü" 1§7¡7¨2¥;Ó/€FÜ˜! ZÓ8Ð8r   c                ód   € V f   \        V4      ^,
          p V ^ 8  d   V \        V4      ,          p V # r]   )rÚ   )r$   r?   s   &&r   Ú_get_flip_dimrê   ù  s.   € à
‚{Ü�%‹j˜1�nˆØ
ˆQ„wØŒs�5‹zÕˆØ€Jr   c                óF  € \         P                  ! \        V P                  4      ) V8*  ;'       d    V\        V P                  4      8  4       \	        WP                  4      p\         P                  ! \        V P                  V,          4      4       \        V P                  V,          4      p\        V P                  P                  RR7      p\         P                  ! V P                  VRR7      V P                  RV ^.V,          ,           V P                  V^,           R ,           4      p\         P                  ! V4       F  pV\        WRV,           R4      ,          pK  	  \         P                  ! WPP                  4      P                  V P                  RR7      p V # )z—
Flips a tensor `x` along the dimension `dim`.

:param x: the first input tensor
:type x: Block
:param dim: the dimension to flip along
:type dim: int
Tr½   rÀ   N)r   r¢   rÚ   r?   rê   r   r   rÂ   r@   rm   r,   ro   rÍ   r¦   )r   r$   r&   ÚstepsrÄ   r¨   r   s   &&     r   rÃ   rÃ     s,  € ô 	×Òœ˜AŸG™G›�}¨Ñ+×BÐB°´c¸!¿'¹'³lÑ0BÔCÜ(¨¯g©gÓ6€DÜ×ÒÔ'¨¯©°­Ó6Ô7Ü! !§'¡'¨$¥-Ó0€Eô  Q§W¡W×%?Ñ%?ÈÔM€FÜ�Š�Q—T‘T˜&¨$�TÓ/°·±¸¸$°À1À#ÈÅ+Õ1MÐPQ×PWÑPWÐX\Ð_`ÕX`ÐXaÐPbÕ1bÓc€AÜ×Ò˜uÖ%ˆØ”˜ !�8 TÓ*Õ*Šñ &ä�Š�QŸ™Ó ×#Ñ# A§G¡G°TÐ#Ó:€AØ€Hr   c                óì   € \         P                  ! W4      p\        VP                  4      ^8X  d   V# \         P                  ! W"P                  RR ^VP                  R,          ,          .,           4      # )a  
Interleaves the values of two tensors along their last dimension. The two tensors must have the same shape.
Equivalent to `tl.join(a, b).reshape(a.shape[:-1] + [2 * a.shape[-1]])`

:param a: The first input tensor.
:type a: Tensor
:param b: The second input tensor.
:type b: Tensor
Néþÿÿÿ)r   ÚjoinrÚ   r?   r,   )r`   ra   Úcs   && r   Ú
interleaverñ     sU   € ô 	�	Š	�!‹€Aä
ˆ1�7‰7ƒ|�qÔàˆô
 �|Š|˜AŸw™w s¨˜|¨q°1·7±7¸2µ;­Ð.?Õ?Ó@Ð@r   c               ó   € V ^8„  d   QhRR/# ©rš   r$   r›   r
   )rœ   s   "r   r�   r�   3  s   € ÷ 8ñ 8�Nñ 8r   c                óÈ   € \         P                  ! V P                  V,          ^8H  4       V P                  V P                  RV V P                  V^,           R ,           4      # )r   N)r   r¢   r?   r,   ©r   r$   s   &&r   Úsqueezerö   2  sJ   € ä×Ò�q—w‘w˜s•| qÑ(Ô)Ø�9‰9�Q—W‘W˜T˜c�] Q§W¡W¨S°1­W¨XÐ%6Õ6Ó7Ð7r   c               ó   € V ^8„  d   QhRR/# ró   r
   )rœ   s   "r   r�   r�   9  s   € ÷ <ñ <�nñ <r   c                ót   € V P                  V P                  R V R,           V P                  VR  ,           4      # )Nr   )r,   r?   rõ   s   &&r   Ú	unsqueezerù   8  s/   € à�9‰9�Q—W‘W˜T˜c�] UÕ*¨Q¯W©W°S°T¨]Õ:Ó;Ð;r   )NFFrF   )NFTF)TF)NFN)NF)r   FN)r   F)NTr]   );Ú
__future__r   Úruntime.jitr   r   Ú r   r   r   r   Úget_int_dtyperÂ   Ú_tensor_member_fnr   Ú_add_math_1arg_docstrr   r   r.   r<   rA   rD   rR   rW   rZ   rb   Ú_add_reduction_docstrr!   r{   r€   rƒ   r…   r‡   r‰   r‹   r�   r–   r"   r    r¦   r©   rª   Ú_add_scan_docstrr¬   r²   r³   rº   rÈ   rÎ   rÒ   ÚCONSTEXPR_0rÝ   râ   rå   rè   rê   rÃ   rñ   rö   rù   r
   r   r   Ú<module>r     s†  ðÝ "ç 1Ý Ý ð
 ñó ðð ñ)ó ð)ñ $ D×$6Ñ$6Ó7€ð ×ÑØñ	"ó ó ð	"ð ×ÑØØ×Ò˜IÓ&ñ"ó 'ó ó ð"ð ×ÑØØ×Ò˜IÓ&ó.ó 'ó ó ð.ð ×ÑØó?ó ó ð?ð ñ$ó ð$ðN ñ	&ó ð	&ð ñ+ó ð+ð ñó ðð ñAó ðAð ñBó ðBð ñó ðð ×ÑØØ×Ò˜IÐ:JØ*IôKóOóKó ó ðOð" ×ÑØØ×Ò˜OÐ;KÔLóó Mó ó ðð ñ ó ð ð ñAó ðAð ñBó ðBð ñó ðð ×ÑØØ×Ò˜IÐ:JØ*IôKóOóKó ó ðOð" ×ÑØØ×Ò˜OÐ;KÔLóó Mó ó ðð
 ñó ðð ñó ðð ×ÑØØ×Ò˜E¨WÔ5öGó 6ó ó ðGð ñó ðð ×ÑØØ×Ò˜IÓ&óGó 'ó ó ðGð ñó ðð ×ÑØØ×Ò˜KÓ(óFó )ó ó ðFð ×ÑØØ×Ò�x¨7Ô3ö	Eó 4ó ó ð	Eð ñó ðð ×ÑØØ×Ò�yÓ!óFó "ó ó ðFð ôó ðð ôó ðð$ ôó ðð* ôó ðð Ø%)ÀÐdh×dtÑdtö %ó ð%ðP Ø"&ÀT×EUÑEUö 8ó ð8ð ö=ó ð=ð6 Ø+/Èd×N^ÑN^ö 9ó ð9ð ñó ðð ×ÑØóó ó ðð. ñAó ðAð, ô8ó ð8ð
 ô<ó ò<r   