+
    –°³j2  ã                  óp   € ^ RI Ht ^ RIHt ^ RIHtHt ^ RIt]'       d   ^RIH	t	 ] ! R R4      4       t
R# )é    )Úannotations)Ú	dataclass)ÚAnyÚTYPE_CHECKINGN©ÚDimEntryc                  ó~   € ] tR t^t$ R]R&   R]R&   R]R&   R]R&   R R	 ltR
 R ltR R lt]RR R ll4       t	Rt
R# )Ú
TensorInfoztorch.Tensor | NoneÚtensorzlist[DimEntry]ÚlevelsÚboolÚ
has_deviceÚbatchedtensorc               ó   € V ^8„  d   QhRR/# )é   ÚreturnÚNone© )Úformats   "Ú[/home/luca/screen-checking/.venv/lib/python3.14/site-packages/functorch/dim/_tensor_info.pyÚ__annotate__ÚTensorInfo.__annotate__   s   € ÷ Jñ J˜tñ Jó    c                	óÒ   a€ ^RI Ho \        ;QJ d)    V3R lV P                   4       F  '       d   K   RM	  RM! V3R lV P                   4       4      '       g   \	        R4      hR# )é   r   c              3  ó<   <"  € T F  p\        VS4      x € K  	  R # 5i©N)Ú
isinstance)Ú.0Úlr   s   & €r   Ú	<genexpr>Ú+TensorInfo.__post_init__.<locals>.<genexpr>   s   øé € Ð@±K¨q”:˜a ×*Ð*³Kùs   ƒFTz%All levels must be DimEntry instancesN)Ú
_dim_entryr   Úallr   ÚAssertionError)Úselfr   s   &@r   Ú__post_init__ÚTensorInfo.__post_init__   sA   ø€ Ý(ç‹sÔ@°D·K²KÓ@�s�sŠsÔ@°D·K²KÓ@×@Ò@Ü Ð!HÓIÐIñ Ar   c               ó   € V ^8„  d   QhRR/# )r   r   Úintr   )r   s   "r   r   r      s   € ÷ +ñ +�cñ +r   c                	ó2   € ^RI Hp V! V P                  4      # )r   )Úndim_of_levels)r#   r,   r   )r&   r,   s   & r   ÚndimÚTensorInfo.ndim   s   € Ý.á˜dŸk™kÓ*Ð*r   c               ó   € V ^8„  d   QhRR/# )r   r   r   r   )r   s   "r   r   r      s   € ÷ 'ñ '˜$ñ 'r   c                	ó   € V P                   R J# r   )r   )r&   s   &r   Ú__bool__ÚTensorInfo.__bool__   s   € Ø�{‰{ $Ð&Ð&r   c               ó(   € V ^8„  d   QhRRRRRRRR/# )r   Úhr   Úensure_batchedr   Úensure_presentr   r
   r   )r   s   "r   r   r   #   s,   € ÷ "5ñ "5Øð"5Ø $ð"5Ø=Að"5à	ñ"5r   c                	óö  € ^RI HpHpHp VP	                  V 4      '       dU   \        V P                  4       V P                  4       V P                  4       V'       d   V P                  4       4      # R4      # VP	                  V 4      '       df   V P                  '       d   V P                  4       MRpV'       d#   V P                  '       d   V P                  4       MRp\        VV! V 4      .RV4      # \        V \        P                  4      '       dI   . p\        V P                  4       ) ^ 4       F  p	VP!                  V! V	4      4       K  	  \        WRV 4      # V'       d   \#        R4      h\        R. RR4      # )r   )ÚDimr   ÚTensorNFTzexpected a tensor object)Ú r8   r   r9   Úcheck_exactr
   Ú_get_tensorÚ_get_levelsÚ_get_has_deviceÚ_get_batchtensorÚis_boundÚ
_get_ranger   ÚtorchÚrangeÚdimÚappendÚ
ValueError)
r4   r5   r6   r8   r   r9   r   Úbatchtensorr   Úis
   &&&       r   ÚcreateÚTensorInfo.create"   s8  € ÷ 	,Ñ+à×Ñ˜a× Ò äØ—‘“Ø—‘“Ø×!Ñ!Ó#ß(6�×"Ñ"Ó$ó	ð ð =Aó	ð ð �_‰_˜Q×Òà'(§z§z z�Q—\‘\”^°tˆFç(6¸1¿:¿:¸:�×"Ñ"Ô$È4ð ô ØÙ˜!“�ØØó	ð ô ˜œ5Ÿ<™<×(Ò(àˆFÜ˜AŸE™E›G˜8 QÖ'�Ø—‘™h q›kÖ*ñ (ä˜a¨¨qÓ1Ð1çÜ Ð!;Ó<Ð<Ü˜d B¨¨tÓ4Ð4r   r   N)TT)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__annotations__r'   r-   r1   ÚstaticmethodrI   Ú__static_attributes__r   r   r   r
   r
      s>   ‡ àÓØÓØÓØ&Ó&õJõ+õ
'ð ö"5ó ô"5r   r
   )Ú
__future__r   Údataclassesr   Útypingr   r   rB   r#   r   r
   r   r   r   Ú<module>rU      s2   ðÝ "å !ß %ã ÷ Ý$ð ÷75ð 75ó ò75r   