
    ig                         d dl mZ d dlmZmZ g dddgg g dgZg dZg ddd	gg g d
gZg dd	dgg g dgZdeee	   ee	   f   fdZ
deee	   ee	   f   dee	   defdZd Zy)    )List)Modelwith_flatten_v2)               )         )r   r   r   r	   r
   r   r   r   )r   r   r	   r   )r   r   	   )r   r	   r
   r   )r   r   
   returnc                  $    t        dt              S )Nmemoize_input)nameforward)r   _memoize_input_forward     u/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/thinc/tests/layers/test_with_flatten.py_memoize_inputr      s    o/EFFr   modelXis_trainc                 v    || j                   d<   dt        t           fd}|D cg c]  }|dz   	 c}|fS c c}w )N
last_inputdYc                 2    | D cg c]  }|dz   	 c}S c c}w )Nr   r   )r   vs     r   backpropz(_memoize_input_forward.<locals>.backprop   s    !"!A"""s   r   )attrsr   int)r   r   r   r"   r!   s        r   r   r      sA     !"EKK#T#Y # aAEx''s   6c                      t        t                     }  | t        d      \  }}|t        k(  sJ | j                  d   j
                  d   t        k(  sJ  |t              t        k(  sJ y )NT)r   r   r   )r   r   INPUTOUTPUTlayersr#   
INPUT_FLATBACKPROP_OUTPUT)r   Yr"   s      r   test_with_flattenr,      s^    N,-E-KAx;;<<?  .*<<<E?o---r   N)typingr   	thinc.apir   r   r&   r)   r'   r*   r$   r   boolr   r,   r   r   r   <module>r0      s     ,	QFB	*%

aVR	+q!fb*5Gd3ic23 G(cDI%&(+/9(@D(.r   