
    i              
           d dl mZmZmZ ddlmZ ddlmZ ddlm	Z	m
Z
 ddlmZ e
Ze	Z ej                  d      deeef   fd	       Zd
eeef   dededeeef   fdZy)    )CallableTuplecast   )registry)Model)Floats2dRagged)	ArrayInfozreduce_sum.v1returnc                  "    t        dt              S )N
reduce_sum)r   forward     h/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/thinc/layers/reduce_sum.pyr   r      s    w''r   modelXris_trainc                       j                   j                  t        t        |j                        |j
                        }|j
                  t        j                  |      dt        dt        f fd}||fS )NdYr   c                 r    j                  |        t        j                  j                  |             S )N)check_consistencyr
   opsbackprop_reduce_sum)r   
array_infolengthsr   s    r   backpropzforward.<locals>.backprop   s/    $$R(eii33B@'JJr   )
r   r   r   r	   datar   r   
from_arrayOutTInT)r   r   r   Yr   r   r   s   `    @@r   r   r      sc    		T(BGG4bjjAAjjG%%a(JKT Kc K h;r   N)typingr   r   r   configr   r   r   typesr	   r
   utilr   r"   r!   layersr   boolr   r   r   r   <module>r*      s    ( (   $  !(E#t)$ ( "(	5d# 	 	 	tX~AV 	r   