o
    i                  
   @   s   d dl mZmZ ddlmZ ddlmZ ddlmZ eZ	eZ
eddee	e
f fdd	Zd
ee	e
f de	dedee
ef fddZdS )    )CallableTuple   )registry)Model)Floats2dzsoftmax_activation.v1returnc                   C   s
   t dtS )Nsoftmax_activation)r   forward r   r   S/home/ubuntu/.local/lib/python3.10/site-packages/thinc/layers/softmax_activation.pyr	      s   
r	   modelXis_trainc                    s0   j j|dd dtdtf fdd} |fS )NF)inplacedYr   c                    s   j j | ddS )N)axis)opsbackprop_softmax)r   Yr   r   r   backprop   s   zforward.<locals>.backprop)r   softmaxOutTInT)r   r   r   r   r   r   r   r
      s   r
   N)typingr   r   configr   r   r   typesr   r   r   layersr	   boolr
   r   r   r   r   <module>   s    .