+
    –°³j“m  ã                   ó4  € ^ RI t ^ RIt^ RIt^ RIt^ RIHt ^ RIt^ RIHt ]P                   ! R R4      4       t	]	! 4       t
RR ltR R lt] P                  R	 4       tR
 t ! R R4      tRR ltRR ltRR ltR tR tR tRR/R ltRR/R ltRR/R ltR# )é    N)ÚAny)Úinfc                   ó@   a € ] tR t^t o ^tRt^t^PtRtV 3R lt	Rt
V tR# )Ú__PrinterOptionséè  Nc                ód   <€ V ^8„  d   Qh/ S[ ;R&   S[;R&   S[ ;R&   S[ ;R&   S[R,          ;R&   # )é   Ú	precisionÚ	thresholdÚ	edgeitemsÚ	linewidthNÚsci_mode)ÚintÚfloatÚbool)ÚformatÚ__classdict__s   "€ÚR/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_tensor_str.pyÚ__annotate__Ú__PrinterOptions.__annotate__   sH   ø‡ ‚ áÑñ ñ Ññ ñ Ññ	 ñ
 Ññ ñ �T�kÑ ò ó    © )Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__r
   r   r   r   r   Ú__annotate_func__Ú__static_attributes__Ú__classdictcell__©r   s   @r   r   r      s%   ø‡ € à€IØ€IØ€IØ€IØ €H÷ ƒ r   r   c                óÚ  € Ve    VR8X  d.   ^\         n        R\         n        ^\         n        ^P\         n        MkVR8X  d.   ^\         n        R\         n        ^\         n        ^P\         n        M7VR8X  d1   ^\         n        \
        \         n        ^\         n        ^P\         n        V e   V \         n        Ve   V\         n        Ve   V\         n        Ve   V\         n        V\         n        R# )až  Set options for printing. Items shamelessly taken from NumPy

Args:
    precision: Number of digits of precision for floating point output
        (default = 4).
    threshold: Total number of array elements which trigger summarization
        rather than full `repr` (default = 1000).
    edgeitems: Number of array items in summary at beginning and end of
        each dimension (default = 3).
    linewidth: The number of characters per line for the purpose of
        inserting line breaks (default = 80). Thresholded matrices will
        ignore this parameter.
    profile: Sane defaults for pretty printing. Can override with any of
        the above options. (any one of `default`, `short`, `full`)
    sci_mode: Enable (True) or disable (False) scientific notation. If
        None (default) is specified, the value is defined by
        `torch._tensor_str._Formatter`. This value is automatically chosen
        by the framework.

Example::

    >>> # Limit the precision of elements
    >>> torch.set_printoptions(precision=2)
    >>> torch.tensor([1.12345])
    tensor([1.12])
    >>> # Limit the number of elements shown
    >>> torch.set_printoptions(threshold=5)
    >>> torch.arange(10)
    tensor([0, 1, 2, ..., 7, 8, 9])
    >>> # Restore defaults
    >>> torch.set_printoptions(profile='default')
    >>> torch.tensor([1.12345])
    tensor([1.1235])
    >>> torch.arange(10)
    tensor([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

NÚdefaultr   ÚshortÚfull)Ú
PRINT_OPTSr
   r   r   r   r   r   )r
   r   r   r   Úprofiler   s   &&&&&&r   Úset_printoptionsr'      sÃ   € ðZ ÒØ�iÔØ#$ŒJÔ Ø#'ŒJÔ Ø#$ŒJÔ Ø#%ŒJÕ Ø˜ÔØ#$ŒJÔ Ø#'ŒJÔ Ø#$ŒJÔ Ø#%ŒJÕ Ø˜ÔØ#$ŒJÔ Ü#&ŒJÔ Ø#$ŒJÔ Ø#%ŒJÔ àÒØ(Œ
ÔØÒØ(Œ
ÔØÒØ(Œ
ÔØÒØ(Œ
ÔØ"„JÖr   c                óF   € V ^8„  d   QhR\         \        \        3,          /# )r	   Úreturn)ÚdictÚstrr   )r   s   "r   r   r   b   s   € ÷ *ñ *œ$œs¤C˜x�.ñ *r   c                 ó6   € \         P                  ! \        4      # )zqGets the current options for printing, as a dictionary that
can be passed as ``**kwargs`` to set_printoptions().
)ÚdataclassesÚasdictr%   r   r   r   Úget_printoptionsr/   b   s   € ô ×ÒœjÓ)Ð)r   c               +  óx   "  € \        4       p\        R/ V B   Rx € \        R/ VB  R#   \        R/ TB  i ; i5i)zuContext manager that temporarily changes the print options.  Accepted
arguments are same as :func:`set_printoptions`.Nr   )r/   r'   )ÚkwargsÚ
old_kwargss   , r   Úprintoptionsr3   i   s9   é € ô "Ó#€JÜÑ�vÒð'ÛäÑ&˜:Ô&øÔÑ&˜:Ó&üs   ‚:™* �:ª7·:c                 óD  € V P                   '       g^   V P                  '       d:   \        P                  P	                  V P
                  4      P                  '       d   V P                  '       d   \        P                  M\        P                  pV P                  VR 7      # ))Údtype)Úis_mpsÚis_xpuÚtorchÚxpuÚget_device_propertiesÚdeviceÚhas_fp64Úis_maiar   ÚdoubleÚto)Útr5   s   & r   Útensor_totyperA   u   sg   € ð �H�HˆHØ——�¤§¡×!@Ñ!@ÀÇÁÓ!J×!S×!SÐ!SØ�y�yˆyô	 	�Šô �\‰\ð 
ð �4‰4�eˆ4ÓÐr   c                   ó2   a € ] tR t^‚t o R tR tR tRtV tR# )Ú
_Formatterc                ó^  € VP                   P                  V n        R V n        RV n        ^V n        \        P                  ! 4       ;_uu_ 4        VP                  R
4      pRRR4       V P                  '       g3   X F*  pT p\        V P
                  \        V4      4      V n        K,  	  R# VP                   \        P                  8X  d    XP                  \        P                  4      p\        P                  ! X\        P                  ! V4      VP!                  ^ 4      ,          4      pVP#                  4       ^ 8X  d   R# VP                   \        P$                  8X  d   VP'                  4       p\)        VP+                  4       4      p\)        VP-                  4       4      p\)        VP                  4       4      pV F'  pV\        P.                  ! V4      8w  g   K   RV n         M	  \0        P                  f&   W‡,          R8„  ;'       g    VR8„  ;'       g    VR8  M\0        P                  V n        V P                  '       dŸ   V P                  '       dS   V FJ  pR\0        P2                   R2P5                  V4      p\        V P
                  \        V4      4      V n        KL  	  R# V F2  pVR p\        V P
                  \        V4      ^,           4      V n        K4  	  R# V P                  '       dS   V FJ  pR\0        P2                   R2P5                  V4      p\        V P
                  \        V4      4      V n        KL  	  R# V FJ  pR\0        P2                   R	2P5                  V4      p\        V P
                  \        V4      4      V n        KL  	  R#   + '       g   i     ELD; i)TFNg     @�@g    „×—Ag-Cëâ6?ú{:.úe}ú.0fúf}éÿÿÿÿ)r5   Úis_floating_pointÚfloating_dtypeÚint_moder   Ú	max_widthr8   Úno_gradÚreshapeÚmaxÚlenÚfloat4_e2m1fn_x2ÚviewÚuint8Úmasked_selectÚisfiniteÚneÚnumelÚfloat8_e8m0fnur   rA   ÚabsÚminÚceilr%   r
   r   )	ÚselfÚtensorÚtensor_viewÚvalueÚ	value_strÚnonzero_finite_valsÚnonzero_finite_absÚnonzero_finite_minÚnonzero_finite_maxs	   &&       r   Ú__init__Ú_Formatter.__init__ƒ   sà  € Ø$Ÿl™l×<Ñ<ˆÔØˆŒØˆŒØˆŒä�]Š]�_�_Ø Ÿ.™.¨Ó,ˆK÷ ð ×"×"Ð"Û$�Ø$˜g�	Ü!$ T§^¡^´S¸³^Ó!D�–ó %ð
 �|‰|œu×5Ñ5Ô5ð *×.Ñ.¬u¯{©{Ó;�ä"'×"5Ò"5ØœUŸ^š^¨KÓ8¸;¿>¹>È!Ó;LÕLó#Ðð #×(Ñ(Ó*¨aÔ/áà�|‰|œu×3Ñ3Ô3ð ':×&?Ñ&?Ó&AÐ#ô "/Ð/B×/FÑ/FÓ/HÓ!IÐÜ!.Ð/A×/EÑ/EÓ/GÓ!HÐÜ!.Ð/A×/EÑ/EÓ/GÓ!HÐã,�ØœEŸJšJ uÓ-Ö-Ø$)�D”MÙñ -ô ×&Ñ&Ò.ð #Õ7¸&Ñ@÷ /ð /Ø%¨Ñ-÷/ð /à%¨Ñ.øä×(Ñ(ð ŒMð �}�}ˆ}ð —=—=�=Û!4˜Ø&*¬:×+?Ñ+?Ð*@ÀÐ$D×$KÑ$KÈEÓ$R˜	Ü),¨T¯^©^¼SÀ»^Ó)L˜žó "5ó "5˜Ø',¨S k˜	Ü),¨T¯^©^¼SÀ»^ÈaÕ=OÓ)P˜žó "5ð
 —=—=�=Û!4˜Ø&*¬:×+?Ñ+?Ð*@ÀÐ$D×$KÑ$KÈEÓ$R˜	Ü),¨T¯^©^¼SÀ»^Ó)L˜žó "5ó "5˜Ø&*¬:×+?Ñ+?Ð*@ÀÐ$D×$KÑ$KÈEÓ$R˜	Ü),¨T¯^©^¼SÀ»^Ó)L˜žó "5÷M �_ˆ_ús   ÁNÎN,	c                ó   € V P                   # ©N)rM   )r]   s   &r   ÚwidthÚ_Formatter.widthÓ   s   € Ø�~‰~Ðr   c                óþ  € V P                   '       dÀ   V P                  '       d2   R V P                   R\        P                   R2P                  V4      pMV P                  '       dG   VR p\        P                  ! V4      '       g&   \        P                  ! V4      '       g
   VR,          pM'R\        P                   R2P                  V4      pMT pV P                  \        V4      ,
          R,          V,           # )z{:Ú.rF   rG   rE   rH   Ú )rK   r   rM   r%   r
   r   rL   ÚmathÚisinfÚisnanrQ   )r]   r`   Úrets   && r   r   Ú_Formatter.formatÖ   s¿   € Ø××ÐØ�}�}ˆ}Ø˜DŸN™NÐ+¨1¬Z×-AÑ-AÐ,BÀ#ÐF×MÑMÈeÓT‘Ø——�Ø˜s˜�ÜŸ
š
 5×)Ò)¬T¯ZªZ¸×->Ò->Ø˜3•J�CøàœZ×1Ñ1Ð2°#Ð6×=Ñ=¸eÓD‘à�GˆCØ—‘¤ S£Õ)¨SÕ0°3Õ6Ð6r   )rK   rL   rM   r   N)	r   r   r   r   rf   rj   r   r   r   r    s   @r   rC   rC   ‚   s   ø‡ € òNMò`÷7ð 7r   rC   c                 ó2  € Vev   \        V P                  V4      p\        V P                  V4      R,           P                  4       pV^ ,          R8X  g   V^ ,          R8X  d	   W4,           # VR,           V,           # VP	                  V P                  4       4      # ©NÚjÚ+Ú-)Ú_scalar_strÚrealÚimagÚlstripr   Úitem)r]   Ú
formatter1Ú
formatter2Úreal_strÚimag_strs   &&&  r   ry   ry   å   s|   € ØÒÜ˜tŸy™y¨*Ó5ˆÜ §	¡	¨:Ó6¸Õ<×DÑDÓFˆà�A�;˜#Ô ¨!¥°Ô!3ØÕ&Ð&à˜c•> HÕ,Ð,à× Ñ  §¡£Ó-Ð-r   c                 óŽ  € VP                  4       ^,           pVe   WTP                  4       ^,           ,          p\        ^\        P                  ! \        P
                  V,
          V,          4      4      pW43R lpV P                  \        P                  8X  d    V P                  \        P                  4      p V'       d   \        P                  '       g   R.pMÒV'       d§   V P                  ^ 4      ^\        P                  ,          8”  d|   V R\        P                   P                  4        U	u. uF
  q—! V	4      NK  	  up	R.,           V \        P                  ) R P                  4        U	u. uF
  q—! V	4      NK  	  up	,           pM$V P                  4        U	u. uF
  q—! V	4      NK  	  pp	\        ^ \        V4      V4       U
u. uF  q¨WªV,            NK  	  pp
V Uu. uF  pRP!                  V4      NK  	  ppRR	RV^,           ,          ,           P!                  V4      ,           R,           # u up	i u up	i u up	i u up
i u upi )
r	   Nc                 ó*  € Ve€   VP                  V P                  4      pVP                  V P                  4      R,           P                  4       pV^ ,          R8X  g   V^ ,          R8X  d	   W4,           # VR,           V,           # VP                  V 4      # ru   )r   rz   r{   r|   )Úvalr~   r   r€   r�   s   &&&  r   Ú_val_formatterÚ#_vector_str.<locals>._val_formatterý   s~   € ØÒ!Ø!×(Ñ(¨¯©Ó2ˆHØ"×)Ñ)¨#¯(©(Ó3°cÕ9×AÑAÓCˆHà˜�{˜cÔ! X¨a¥[°CÔ%7ØÕ*Ð*à #•~¨Õ0Ð0à×$Ñ$ SÓ)Ð)r   ú...z ...ú, Ú[rn   Ú]ú,
)rj   rP   ro   Úfloorr%   r   r5   r8   rR   rS   rT   r   ÚsizeÚtolistÚrangerQ   Újoin)r]   ÚindentÚ	summarizer~   r   Úelement_lengthÚelements_per_liner…   Údatar„   ÚiÚ
data_linesÚlineÚliness   &&&&&         r   Ú_vector_strrš   ò   sì  € à×%Ñ%Ó'¨!Õ+€NØÒà×*Ñ*Ó,¨qÕ0Õ0ˆäØ	Œ4�:Š:”z×+Ñ+¨fÕ4¸ÕHÓIóÐð (2ô 
*ð ‡z�z”U×+Ñ+Ô+ð �y‰yœŸ™Ó%ˆçœ×-×-Ð-àˆw‰ß	�t—y‘y “| a¬*×*>Ñ*>Õ&>Ô>à,0Ð1G´:×3GÑ3GÐ,H×,OÑ,OÔ,QÓRÑ,Q Sˆ^˜CÖ Ñ,QÑRØˆhõà.2´J×4HÑ4HÐ3HÐ3JÐ.K×.RÑ.RÔ.TÓUÑ.T sˆ~˜cÖ"Ñ.TÑUõVñ 	ð 04¯{©{¬}Ó=©}¨�˜sÖ#©}ˆÐ=ô 27°q¼#¸d»)ÐEVÔ1WóÙ1W¨AˆQÐ&Õ&Ó'Ñ1Wð ð ñ *4Ó4© ˆT�Y‰Y�tŽ_©€EÐ4Ø�*˜s f¨q¥jÕ1Õ1×7Ñ7¸Ó>Õ>ÀÕDÐDùò SùâUùò >ùòùò 5s   Ä1H.Å/H3ÆH8ÇH=ÇIc                 óf  € V P                  4       pV^ 8X  d   \        WV4      # V^8X  d   \        WW#V4      # V'       dØ   V P                  ^ 4      ^\        P
                  ,          8”  d­   \        \        P
                  4       Uu. uF  p\        W,          V^,           W#V4      NK   	  upR.,           \        \        V 4      \        P
                  ,
          \        V 4      4       Uu. uF  p\        W,          V^,           W#V4      NK   	  up,           pMB\        V P                  ^ 4      4       Uu. uF  p\        W,          V^,           W#V4      NK   	  ppRRV^,
          ,          ,           RV^,           ,          ,           P                  V4      pRV,           R,           # u upi u upi u upi )r   r‡   Ú,Ú
rn   r‰   rŠ   )
Údimry   rš   r�   r%   r   r�   Ú_tensor_str_with_formatterrQ   r�   )	r]   r‘   r’   r~   r   rž   r–   ÚslicesÚ
tensor_strs	   &&&&&    r   rŸ   rŸ   '  s�  € Ø
�(‰(‹*€Cà
ˆa„xÜ˜4¨ZÓ8Ð8à
ˆa„xÜ˜4¨À
ÓKÐKç�T—Y‘Y˜q“\ A¬
×(<Ñ(<Õ$<Ô<ô œz×3Ñ3Ô4ó	ñ 5�Aô +Ø•G˜V a�Z¨À
öñ 5ñ	ð ˆgõô œs 4›y¬:×+?Ñ+?Õ?ÄÀTÃÔKó	ñ L�Aô +Ø•G˜V a�Z¨À
öñ Lñ	õñ 	ô( ˜4Ÿ9™9 Q›<Ô(ó	
ñ )�ô 'Ø•˜ !� Y¸Jöñ )ð	 	ð 
ð ˜  a¥Õ(Õ(¨3°&¸1µ*Õ+=Õ=×CÑCÀFÓK€JØ�Õ˜cÕ!Ð!ùò/ùòùò
s   Â$F$Ã)$F)Ä3$F.c                 óR  € V P                  4       ^ 8X  d   R# V P                  4       \        P                  8„  pV P                  4       '       d   V P	                  4       p V P                  4       '       d   V P                  4       p V P                  \        P                  \        P                  \        P                  \        P                  39   d   V P                  4       p V P                  P                  '       d‚   V P                  4       p \!        V'       d   \#        V P$                  4      MV P$                  4      p\!        V'       d   \#        V P&                  4      MV P&                  4      p\)        WW#V4      # \!        V'       d   \#        V 4      MT 4      p\)        WW%4      # )r   ú[])rX   r%   r   Ú_is_zerotensorÚcloneÚis_negÚresolve_negr5   r8   Úfloat8_e5m2Úfloat8_e5m2fnuzÚfloat8_e4m3fnÚfloat8_e4m3fnuzÚhalfÚ
is_complexÚresolve_conjrC   Úget_summarized_datarz   r{   rŸ   )r]   r‘   r’   Úreal_formatterÚimag_formatterÚ	formatters   &&    r   Ú_tensor_strr³   L  s9  € Ø‡z�zƒ|�qÔÙà—
‘
“œz×3Ñ3Ñ3€Ià×Ñ×ÒØ�z‰z‹|ˆð ‡{�{‡}‚}Ø×ÑÓ!ˆð ‡z�zÜ×ÑÜ×ÑÜ×ÑÜ×Ñð	ô ð �y‰y‹{ˆà‡z�z××Ðà× Ñ Ó"ˆÜ#ß.7Ô §	¡	Ô*¸T¿Y¹Yó
ˆô $ß.7Ô §	¡	Ô*¸T¿Y¹Yó
ˆô *Ø˜)°^ó
ð 	
ô ¿IÔ2°4Ô8È4ÓPˆ	Ü)¨$¸	ÓMÐMr   c                 óÊ  € V .p\        V 4      V P                  R 4      ,
          ^,           pV F‘  p\        V4      pV'       g#   WW,           ^,           \        P                  8”  d3   VP	                  RRV,          ,           V,           4       W',           pRpKj  VP	                  RV,           4       WW^,           ,          pK“  	  VP	                  R4       RP                  V4      # )r�   r‹   rn   Frˆ   Ú)Ú )rQ   Úrfindr%   r   Úappendr�   )r¡   Úsuffixesr‘   Úforce_newlineÚtensor_strsÚlast_line_lenÚsuffixÚ
suffix_lens   &&&&    r   Ú_add_suffixesr¿   s  s¸   € Ø�,€KÜ˜
“O j×&6Ñ&6°tÓ&<Õ<¸qÕ@€MÛˆÜ˜“[ˆ
ß˜MÕ6¸Õ:¼Z×=QÑ=QÔQØ×Ñ˜u s¨V¥|Õ3°fÕ<Ô=Ø"Õ/ˆMØ!ŠMà×Ñ˜t f�}Ô-Ø¨!�^Õ+ŠMñ ð ×Ñ�sÔØ�7‰7�;ÓÐr   c                 ó¾  € V P                  4       pV^ 8X  d   V # V^8X  di   V P                  ^ 4      ^\        P                  ,          8”  d<   \        P
                  ! V R\        P                   V \        P                  ) R 34      # V # \        P                  '       g(   V P                  ^ .V P                  4       ,          4      # V P                  ^ 4      ^\        P                  ,          8”  d¬   \        \        P                  4       Uu. uF  q V,          NK  	  pp\        \        V 4      \        P                  ,
          \        V 4      4       Uu. uF  q V,          NK  	  pp\        P                  ! W4,            Uu. uF  p\        V4      NK  	  up4      # \        P                  ! V  Uu. uF  p\        V4      NK  	  up4      # u upi u upi u upi u upi )r   N)rž   r�   r%   r   r8   ÚcatÚ	new_emptyr�   rQ   Ústackr¯   )r]   rž   r–   ÚstartÚendÚxs   &     r   r¯   r¯   ƒ  sn  € Ø
�(‰(‹*€CØ
ˆa„xØˆØ
ˆa„xØ�9‰9�Q‹<˜!œj×2Ñ2Õ2Ô2Ü—9’9ØÐ,œ
×,Ñ,Ð-¨t´Z×5IÑ5IÐ4IÐ4KÐ/LÐMóð ð ˆKÜ××ÐØ�~‰~˜q˜c D§H¡H£JÕ.Ó/Ð/Ø	�‰�1‹˜œJ×0Ñ0Õ0Ô	0Ü"'¬
×(<Ñ(<Ô"=Ó>Ñ"=˜Q�a—�Ñ"=ˆÐ>Ü %¤c¨$£i´*×2FÑ2FÕ&FÌÈDË	Ô RÓSÑ R˜1�A�wˆwÑ RˆÐSÜ�{Š{¸U¾[ÓJ¹[°qÔ/°Ö2¹[ÑJÓKÐKä�{Š{¹DÓA¹D°qÔ/°Ö2¹DÑAÓBÐBùò	 ?ùÚSùÚJùâAs   ÄGÅGÆGÆ0GÚtensor_contentsc          	      ó   a!a"€ \         P                  P                  P                  V 4      '       d   \	        WR 7      # \        V 4      \         P                  J ;'       g&    \        V 4      \         P                  P                  J pV P                  '       d   RpM#V'       d   RpM\        V 4      P                   R2p\        V4      o!. pVRJpV'       d   Tp\         P                  P                  P                  V 4      w  rxVP                  P
                  \         P                  P!                  4       8w  gn   VP                  P
                  R8X  d8   \         P"                  P%                  4       VP                  P&                  8w  g   VP                  P
                  R8X  d3   VP)                  R\+        VP                  4      ,           R,           4       VP                  P
                  R/9   d   VP-                  R	4      p\         P.                  ! 4       \         P0                  8X  d   \         P2                  M\         P4                  p	VP6                  \         P.                  ! 4       V	\         P8                  \         P:                  39   p
VP<                  '       Ed   TP)                  R
\+        \?        VP@                  4      4      ,           4       ^ RI!H"p VPF                  ;'       g	    V! V4      pV'       g0   VP)                  R\+        VPI                  4       4      ,           4       V
'       g,   VP)                  R\+        VP6                  4      ,           4       V'       EgT   RpVPK                  4       PM                  4       pV'       d   RpM\O        VS!\        V4      ,           4      pV'       g   VPQ                  4       ^ 8X  d-   VR\+        \?        VP@                  4      4      ,           ,          pRpVPS                  4       PM                  4       pV'       d   RpM\O        VS!\        V4      ,           4      pV'       g   VPQ                  4       ^ 8X  d-   VR\+        \?        VP@                  4      4      ,           ,          pVV,           R,           RS!,          ,           V,           V,           R,           pE	MÞVPT                  \         PV                  \         PX                  \         PZ                  \         P\                  09   Ed-   ^ RI!H"p VP)                  R
\+        \?        VP@                  4      4      ,           4       VPF                  ;'       g	    V! V4      pV'       g0   VP)                  R\+        VPI                  4       4      ,           4       V
'       g,   VP)                  R\+        VP6                  4      ,           4       V'       Egb   \         PV                  \         P                  P^                  \         P                  P`                  3\         PX                  \         P                  Pb                  \         P                  Pd                  3\         PZ                  \         P                  P^                  \         P                  P`                  3\         P\                  \         P                  Pb                  \         P                  Pd                  3/VPT                  ,          w  ppVPT                  \         PV                  \         PZ                  09   d   RRppMRRppRVR,           R2pV! V4      PM                  4       pV'       d   RpM\O        VS!\        V4      ,           4      pVPQ                  4       ^ 8X  g	   V'       d-   VR\+        \?        VP@                  4      4      ,           ,          pVR,           R2pV! V4      PM                  4       pV'       d   RpM\O        VS!\        V4      ,           4      pVPQ                  4       ^ 8X  g	   V'       d-   VR\+        \?        VP@                  4      4      ,           ,          pRpVPg                  4       PM                  4       pV'       d   RpM\O        VS!\        V4      ,           4      pVPQ                  4       ^ 8X  g	   V'       d-   VR\+        \?        VP@                  4      4      ,           ,          pVV,           R,           RS!,          ,           V,           V,           R,           RS!,          ,           V,           V,           R,           pEMdVPh                  '       EdV   VP)                  R
\+        \?        VP@                  4      4      ,           4       V
'       g,   VP)                  R\+        VP6                  4      ,           4       VP)                  R\+        VPk                  4       4      ,           4       VPk                  4       \         Pl                  8X  g$   VPk                  4       \         Pn                  8X  d`   VP)                  R\+        VPq                  4       4      ,           4       VP)                  R\+        VPs                  4       4      ,           4       MöVPk                  4       \         Pt                  8X  gG   VPk                  4       \         Pv                  8X  g$   VPk                  4       \         Px                  8X  dŽ   VP)                  R\+        VP{                  4       4      ,           4       VP)                  R\+        VP}                  4       4      ,           4       VP)                  R\+        VP                  4       4      ,           4       V'       g   \O        VP�                  4       S!4      pEMüVP                  '       db   V'       gX   R o"RPƒ                  V!V"3R  l\         P„                  P†                  Pˆ                  P‹                  V^ 4       4       4      pR!V R"2pEM‰\         PŒ                  ! V4      '       d$   R#p\�        \         P�                  ! V4      4      pEMJ^ RI!H"p VPF                  '       g   V! V4      '       d�   VP)                  R
\+        \?        VP@                  4      4      ,           4       VP6                  \         P.                  ! 4       8w  d,   VP)                  R\+        VP6                  4      ,           4       V'       g   RpEM•VPQ                  4       ^ 8X  d¶   VP<                  '       g¤   VP“                  4       ^8w  d5   VP)                  R
\+        \?        VP@                  4      4      ,           4       VP6                  \         P.                  ! 4       8w  d,   VP)                  R\+        VP6                  4      ,           4       V'       g   R$pMË\”        P–                  '       g5   VP)                  R
\+        \?        VP@                  4      4      ,           4       V
'       g,   VP)                  R\+        VP6                  4      ,           4       V'       gG   VPT                  \         P˜                  8w  d   \O        VP›                  4       S!4      pM\O        VS!4      pVPT                  \         P˜                  8w  d,   VP)                  R%\+        VPT                  4      ,           4       Rp V Pœ                  pVfH   XeD   \        V4      P                  pVR'8X  d(   VP¡                  4       P£                  R(^4      R0,          pVe   VP)                  R)V R*24       M#V P¤                  '       d   VP)                  R+4       Ve   VP)                  R,V 24       \§        VX,           VS!VP<                  R-7      p \©        V\         P                  P                  4      '       d   V'       g   R.V  R2p V #   \ž         d    R&p EL
i ; i)1©rÇ   znested_tensor(ztensor(Ú(NÚcudaÚmpszdevice='Ú'Úcpuzsize=)Úis_fake_tensorznnz=zdtype=zindices=tensor(r‡   z, size=zvalues=tensor(z),
rn   rµ   ÚrowÚcolumnÚc:Né   Nz_indices=tensor(zquantization_scheme=zscale=zzero_point=zaxis=c                 óP   € R P                  R V P                  R 4       4       4      # )r�   c              3   ó,   "  € T F
  pR V 2x € K  	  R# 5i)z  Nr   )Ú.0r˜   s   & r   Ú	<genexpr>Ú4_str_intern.<locals>.indented_str.<locals>.<genexpr>N  s   é € Ð G¹° 2 d V¥»ùs   ‚)r�   Úsplit)Úsr‘   s   &&r   Úindented_strÚ!_str_intern.<locals>.indented_strM  s   € Ø—y‘yÑ G¸¿¹À¼Ó GÓGÐGr   r‹   c              3   óV   <"  € T F  pS! \        V4      S^,           4      x € K   	  R# 5i)é   N)r+   )rÖ   r@   r‘   rÛ   s   & €€r   r×   Ú_str_intern.<locals>.<genexpr>P  s*   øé € ð á;�Añ œS ›V V¨a¥Z×0Ð0Û;ùs   ƒ&)z[
z
]z_to_functional_tensor(r£   zlayout=ÚInvalidÚCppFunctionz::z	grad_fn=<Ú>zrequires_grad=Trueztangent=)rº   z
Parameter()ÚxlaÚlazyÚipuÚmtiarI   )Ur8   Ú_CÚ
_functorchÚis_functorch_wrapped_tensorÚ_functorch_wrapper_str_internÚtypeÚTensorÚnnÚ	ParameterÚ	is_nestedr   rQ   ÚautogradÚ
forward_adÚunpack_dualr;   Ú_get_default_devicerË   Úcurrent_deviceÚindexr¸   r+   r?   Úget_default_dtyper>   ÚcdoubleÚcfloatr5   Úint64r   Ú	is_sparseÚtupleÚshapeÚtorch._subclasses.fake_tensorrÏ   Úis_metaÚ_nnzÚ_indicesÚdetachr³   rX   Ú_valuesÚlayoutÚ
sparse_csrÚ
sparse_cscÚ
sparse_bsrÚ
sparse_bscÚcrow_indicesÚcol_indicesÚccol_indicesÚrow_indicesÚvaluesÚis_quantizedÚqschemeÚper_tensor_affineÚper_tensor_symmetricÚq_scaleÚq_zero_pointÚper_channel_affineÚper_channel_symmetricÚ per_channel_affine_float_qparamsÚq_per_channel_scalesÚq_per_channel_zero_pointsÚq_per_channel_axisÚ
dequantizer�   ÚopsÚatenÚunbindr   Ú_is_functional_tensorÚreprÚ_from_functional_tensorrž   r%   r   ÚstridedÚto_denseÚgrad_fnÚRuntimeErrorÚnameÚrsplitÚrequires_gradr¿   Ú
isinstance)#ÚinprÇ   Úis_plain_tensorÚprefixr¹   Úcustom_contents_providedr¡   r]   ÚtangentÚ_default_complex_dtypeÚhas_default_dtyperÏ   rþ   Úindices_prefixÚindicesÚindices_strÚvalues_prefixr  Ú
values_strÚcompressed_indices_methodÚplain_indices_methodÚcdimnameÚpdimnameÚcompressed_indices_prefixÚcompressed_indicesÚcompressed_indices_strÚplain_indices_prefixÚplain_indicesÚplain_indices_strÚstrsÚgrad_fn_namer"  Ústring_reprr‘   rÛ   s#   &$                               @@r   Ú_str_internrA  ˜  sY  ù€ Ü‡x�x×Ñ×6Ñ6°s×;Ò;Ü,¨SÔRÐRÜ˜3“i¤5§<¡<Ð/×RÐR´4¸³9ÄÇÁ×@RÑ@RÐ3R€OØ
‡}‡}€}Ø!‰ß	Ø‰ä˜“I×&Ñ&Ð' qÐ)ˆÜ�‹[€FØ€HØ.°dÐ:ÐßØ$ˆ
ô
 —N‘N×-Ñ-×9Ñ9¸#Ó>�M€Dð 	�‰×ÑœEŸH™H×8Ñ8Ó:Ô:à�K‰K×Ñ Ô&Ü—
‘
×)Ñ)Ó+¨t¯{©{×/@Ñ/@Ô@à�K‰K×Ñ Ô%à�‰˜
¤S¨¯©Ó%5Õ5¸Õ;Ô<ð
 ‡{�{×ÑÐ9Ô9Ø�w‰w�u‹~ˆô ×0Ò0Ó2´e·l±lÔBŒ�ŠÌÏÉð ð Ÿ
™
Ü×ÒÓ!ØÜ�‰Ü�
‰
ð	'ñ Ðð ‡~‡~�~Ø�‰˜¤#¤e¨D¯J©JÓ&7Ó"8Õ8Ô9Ý@à—,‘,×6Ð6¡.°Ó"6ˆßØ�O‰O˜F¤S¨¯©«Ó%5Õ5Ô6ß Ø�O‰O˜H¤s¨4¯:©:£Õ6Ô7ß'Ð'Ø.ˆNØ—m‘m“o×,Ñ,Ó.ˆGßØ#‘ä)¨'°6¼CÀÓ<OÕ3OÓP�ß˜'Ÿ-™-›/¨QÔ.Ø˜y¬3¬u°W·]±]Ó/CÓ+DÕDÕD�Ø,ˆMØ—\‘\“^×*Ñ*Ó,ˆFßØ"‘
ä(¨°¼#¸mÓ:LÕ1LÓM�
ß˜&Ÿ,™,›.¨AÔ-Ø˜i¬#¬e°F·L±LÓ.AÓ*BÕBÕB�
àØõàõð ˜•,õð  õ	 ð
 õð õð ùð 
�‰Ü×ÑÜ×ÑÜ×ÑÜ×Ñð	õ 
õ 	Aà�‰˜¤#¤e¨D¯J©JÓ&7Ó"8Õ8Ô9Ø—,‘,×6Ð6¡.°Ó"6ˆßØ�O‰O˜F¤S¨¯©«Ó%5Õ5Ô6ß Ø�O‰O˜H¤s¨4¯:©:£Õ6Ô7ß'Ð'ä× Ñ ¤5§<¡<×#<Ñ#<¼e¿l¹l×>VÑ>VÐ"WÜ× Ñ ¤5§<¡<×#<Ñ#<¼e¿l¹l×>VÑ>VÐ"WÜ× Ñ ¤5§<¡<×#<Ñ#<¼e¿l¹l×>VÑ>VÐ"WÜ× Ñ ¤5§<¡<×#<Ñ#<¼e¿l¹l×>VÑ>VÐ"Wð	?ð
 �k‰kõ?Ñ;Ð%Ð';ð �{‰{œu×/Ñ/´×1AÑ1AÐBÔBØ%*¨H˜(�˜(à%-¨u˜(�Ø*+¨H°R­L¨>Ð9IÐ(JÐ%Ù!:¸4Ó!@×!GÑ!GÓ!IÐßØ).Ñ&ä)4Ø&¨´Ð5NÓ1OÕ(Oó*Ð&ð "×'Ñ'Ó)¨QÔ.·'Ø&¨)´cÜÐ,×2Ñ2Ó3ó7õ +õ Ð&ð '/¨r¥l ^Ð3CÐ#DÐ Ù0°Ó6×=Ñ=Ó?ˆMßØ$)Ñ!ä$/Ø! 6¬CÐ0DÓ,EÕ#Eó%Ð!ð ×"Ñ"Ó$¨Ô)¯WØ! Y´´U¸=×;NÑ;NÓ5OÓ1PÕ%PÕPÐ!Ø,ˆMØ—[‘[“]×)Ñ)Ó+ˆFßØ"‘
ä(¨°¼#¸mÓ:LÕ1LÓM�
Ø�|‰|‹~ Ô"§gØ˜i¬#¬e°F·L±LÓ.AÓ*BÕBÕB�
à)Ø(õ)àõð ˜•,õð 'õ	'ð
 $õ$ð õð ˜•,õð  õ ð õ	ð õ
ð ùð 
×	×	Ñ	Ø�‰˜¤#¤e¨D¯J©JÓ&7Ó"8Õ8Ô9ß Ø�O‰O˜H¤s¨4¯:©:£Õ6Ô7Ø�‰Ð.´°T·\±\³^Ó1DÕDÔEà�L‰L‹Nœe×5Ñ5Ô5Ø�|‰|‹~¤×!;Ñ!;Ô;à�O‰O˜H¤s¨4¯<©<«>Ó':Õ:Ô;Ø�O‰O˜M¬C°×0AÑ0AÓ0CÓ,DÕDÕEà�L‰L‹Nœe×6Ñ6Ô6Ø�|‰|‹~¤×!<Ñ!<Ô<Ø�|‰|‹~¤×!GÑ!GÔGà�O‰O˜H¤s¨4×+DÑ+DÓ+FÓ'GÕGÔHØ�O‰O˜M¬C°×0NÑ0NÓ0PÓ,QÕQÔRØ�O‰O˜G¤c¨$×*AÑ*AÓ*CÓ&DÕDÔEß'Ü$ T§_¡_Ó%6¸Ó?ˆJùØ	��ˆß'òHð —:‘:õ äŸ™Ÿ™×.Ñ.×2Ñ2°4¸Ô;óó ˆDð ˜t˜f C˜ˆJùÜ	×	$Ò	$ T×	*Ò	*Ø)ˆÜœ%×7Ò7¸Ó=Ó>Š
õ 	Aà�<�<ˆ<™>¨$×/Ò/Ø�O‰O˜G¤c¬%°·
±
Ó*;Ó&<Õ<Ô=Ø�z‰zœU×4Ò4Ó6Ô6Ø—‘ ¬3¨t¯z©z«?Õ :Ô;÷ ,Ø"�
ùà�z‰z‹|˜qÔ ¨¯¯¨à—8‘8“: ”?Ø—O‘O G¬c´%¸¿
¹
Ó2CÓ.DÕ$DÔEð —:‘:¤×!8Ò!8Ó!:Ô:Ø—O‘O H¬s°4·:±:«Õ$>Ô?ß/Ø!%�Jøä!×+×+Ð+Ø—O‘O G¬c´%¸¿
¹
Ó2CÓ.DÕ$DÔEç(Ø—O‘O H¬s°4·:±:«Õ$>Ô?ç/Ø—{‘{¤e§m¡mÔ3Ü%0°·±³À&Ó%I™
ä%0°°vÓ%>˜
à‡{�{”e—m‘mÔ#Ø�‰˜	¤C¨¯©Ó$4Õ4Ô5ð €Lð!Ø—+‘+ˆð Ò Ò 3ä˜G“}×-Ñ-ˆØ˜=Ô(à"Ÿ<™<›>×0Ñ0°°qÓ9¸"Õ=ˆLàÒØ�‰˜) L >°Ð3Õ4Ø	×	×	Ð	Ø�‰Ð,Ô-àÒØ�‰˜( 7 )Ð,Ô-äØ�ÕØØØ—n‘nô	€Kô �$œŸ™×*Ñ*×+Ò+·OØ" ; -¨qÐ1ˆàÐøôG ô !ð !‹ð	!ús   û,2 ÿ2A@Á@A@c                óz  € \         P                  P                  P                  V 4      pVR8X  d   \	        R4      h\         P                  P                  P                  V 4      '       d   \         P                  ! V 4       \         P                  P                  P                  V 4      p\        V4      p\        P                  ! VR4      p\         P                  P                  P                  V 4      '       dH   \         P                  P                  P                  V 4      pVR8X  d   \	        R4      hRV RV RV R2# \         P                  P                  P                  V 4      '       d
   RV RV R2# \         P                  P                  P                  V 4      '       d
   RV R	V R
2# \        R4      h)rÞ   z+expected functorch level to be >= 0, got -1z+expected batch dimension to be >= 0, got -1zBatchedTensor(lvl=z, bdim=z	, value=
z
)zGradTrackingTensor(lvl=zFunctionalTensor(lvl=z
, value=\
rµ   z8We don't know how to print this, please file us an issuerI   z    )r8   rç   rè   Úmaybe_get_levelÚAssertionErrorÚis_functionaltensorÚ_syncÚget_unwrappedr  Útextwrapr‘   Úis_batchedtensorÚmaybe_get_bdimÚis_gradtrackingtensorÚ
ValueError)r^   rÇ   Úlevelr`   Ú
value_reprÚindented_value_reprÚbdims   &$     r   rê   rê   «  sd  € Ü�H‰H×Ñ×/Ñ/°Ó7€EØ�„{ÜÐJÓKÐKä‡x�x×Ñ×.Ñ.¨v×6Ò6ô 	�Š�FÔä�H‰H×Ñ×-Ñ-¨fÓ5€EÜ�e“€Jä"Ÿ/š/¨*°gÓ>ÐÜ‡x�x×Ñ×+Ñ+¨F×3Ò3Ü�x‰x×"Ñ"×1Ñ1°&Ó9ˆØ�2Œ:Ü Ð!NÓOÐOà    w¨t¨f°JÐ?RÐ>SÐSVÐWð	
ô ‡x�x×Ñ×0Ñ0°×8Ò8Ø(¨¨¨zÐ:MÐ9NÈcÐRÐRÜ‡x�x×Ñ×.Ñ.¨v×6Ò6Ø& u g¨\¸*¸ÀQÐGÐGä
ÐOÓ
PÐPr   c          	      ó~  € \         P                  ! 4       ;_uu_ 4        \         P                  P                  P	                  4       ;_uu_ 4        \         P
                  P                  4       p\        WR 7      uuRRR4       uuRRR4       #   + '       g   i     M; iRRR4       R#   + '       g   i     R# ; i)rÉ   N)r8   rN   ÚutilsÚ_python_dispatchÚ_disable_current_modesrç   Ú_DisableFuncTorchrA  )r]   rÇ   Úguards   &$ r   Ú_strrW  È  s\   € Ü	�Š��œ%Ÿ+™+×6Ñ6×MÑM×OÕOÜ—‘×*Ñ*Ó,ˆÜ˜4ÔA÷ PÑO�Š×O×Oú���Šús#   ž1B+Á*B	Á9
B+ÂBÂB+Â+B<	)NNNNNNri   )Ú
contextlibr-   ro   rH  Útypingr   r8   r   Ú	dataclassr   r%   r'   r/   Úcontextmanagerr3   rA   rC   ry   rš   rŸ   r³   r¿   r¯   rA  rê   rW  r   r   r   Ú<module>r\     sÊ   ðã Û Û Û Ý ã Ý ð ×Ñ÷!ð !ó ð!ñ Ó€
ôF#õR*ð ×Ññ'ó ð'ò
÷`7ñ `7ôF
.ô/Eôj""òJ$NòN ò Cð*P¨ô PðfQ¸Tô Qð:B $ö Br   