
    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_mean.v1returnc                  "    t        dt              S )Nreduce_mean)r   forward     i/var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/thinc/layers/reduce_mean.pyr   r      s    ((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_mean)r   
array_infolengthsr   s    r   backpropzforward.<locals>.backprop   s/    $$R(eii44RA7KKr   )
r   r   r   r	   datar   r   
from_arrayOutTInT)r   r   r   Yr   r   r   s   `    @@r   r   r      sc    		d8RWW5rzzBAjjG%%a(JLT Lc L h;r   N)typingr   r   r   configr   r   r   typesr	   r
   utilr   r"   r!   layersr   boolr   r   r   r   <module>r*      s    ( (   $  !")U39% ) #)
5d# 
 
 
tX~AV 
r   