o i @sNddlZddlZddlmZddlmZddlZ Gdddej Z ddZ dS)Ncs0eZdZdZfddZddZddZZS)CTCzChainer implementation of ctc layer. Args: odim (int): The output dimension. eprojs (int | None): Dimension of input vectors from encoder. dropout_rate (float): Dropout rate. csVtt|||_d|_|t|||_WddS1s$wYdS)N) superr__init__ dropout_rateloss init_scopeLLinearctc_lo)selfodimeprojsr __class__S/home/ubuntu/.local/lib/python3.10/site-packages/espnet/nets/chainer_backend/ctc.pyrs  "z CTC.__init__c Csd|_dd|D}dd|D}|jtjt||jddd}tj|dd }tj|d d }t|j j |t j d }t|j j |t j d }t |jjd t|jt |jjdt|jt||d|||_t dt|jj|jS)a^CTC forward. Args: hs (list of chainer.Variable | N-dimension array): Input variable from encoder. ys (list of chainer.Variable | N-dimension array): Input variable of decoder. Returns: chainer.Variable: A variable holding a scalar value of the CTC loss. NcSg|]}|jdqSrshape.0xrrr )z CTC.__call__..cSrrrrrrrr*r)ratio n_batch_axes)axis)padding)dtypez input lengths: z output lengths: rz ctc loss:)rr Fdropout pad_sequencerseparatechainerVariablexparraynpint32logginginfor__name__strdata%connectionist_temporal_classification) r hsysilensolensy_haty_true input_length label_lengthrrr__call__s*  z CTC.__call__cCs4|jt|dd}t|d|jd|jS)zLog_softmax of frame activations. Args: hs (list of chainer.Variable | N-dimension array): Input variable from encoder. Returns: chainer.Variable: A n-dimension float array. rrr!)r r$r& log_softmaxreshaper)r r4r8rrrr=Gs zCTC.log_softmax)r0 __module__ __qualname____doc__rr<r= __classcell__rrrrr s  ,rcCs:|j}|dkrtdt||j|j}|Std|)zReturn the CTC layer corresponding to the args. Args: args (Namespace): The program arguments. odim (int): The output dimension. Returns: The CTC module. builtinz Using chainer CTC implementationzctc_type must be "builtin": {})ctc_typer.r/rr r ValueErrorformat)argsr rDctcrrrctc_forVs   rI) r.r(chainer.functions functionsr$ chainer.linkslinksrnumpyr,ChainrrIrrrrs   M