
    h              
           S SK JrJr  SSKJr  SSKJr  SSKJr  \r	\r
\R                  " S5      S\\	\
4   4S j5       rS	\\	\
4   S
\	S\S\\
\4   4S jrg)    )CallableTuple   )registry)Model)Floats2dzsoftmax_activation.v1returnc                  "    [        S[        5      $ )Nsoftmax_activation)r   forward     Y/home/james-whalen/.local/lib/python3.13/site-packages/thinc/layers/softmax_activation.pyr   r      s    %w//r   modelXis_trainc                 l   ^ ^ T R                   R                  USS9mS[        S[        4UU 4S jjnTU4$ )NF)inplacedYr	   c                 :   > TR                   R                  TU SS9$ )N)axis)opsbackprop_softmax)r   Yr   s    r   backpropforward.<locals>.backprop   s    yy))!Rb)99r   )r   softmaxOutTInT)r   r   r   r   r   s   `   @r   r   r      s@    		!U+A:T :c : : h;r   N)typingr   r   configr   r   r   typesr   r    r   layersr   boolr   r   r   r   <module>r&      s|    "    
()0E#t), 0 *05d#  t dHn@U r   