o i @sBddlZ d dejdejdejdejdeded ejfd d ZdS) N log_probstargets input_lengthstarget_lengthsblank ignore_idreturnc Cs||||k<|\}}}tj||jd} tjtjt|||fddjddt|ddddf|fdd} tjtjddgg|jd |d| ddddf| dddd fkfdd} tj t d |j|j d } d} | | d}tj ||f| |j|j d }|ddd |f|dd| d f<|| d | dddff|dd| df<tj|||f|j|j d }td|D]P}t|ddddf|ddddft| |dddd f| f}|jd d\}}|dd|fd| ||dd| df<||dd|| df<q|dtj| |dd| |dfdd}tj||f|jtjd }| |dd|jdd|| |df<t|dd dD]"}|dd|f}|| ||f}|dd|df||7<qK| jd|| jd d d}|S)a-Align a CTC label sequence to an emission. Args: log_probs (Tensor): log probability of CTC emission output. Tensor of shape `(B, T, C)`. where `B` is the batch size, `T` is the input length, `C` is the number of characters in alphabet including blank. targets (Tensor): Target sequence. Tensor of shape `(B, L)`, where `L` is the target length. input_lengths (Tensor): Lengths of the inputs (max value must each be <= `T`). 1-D Tensor of shape `(B,)`. target_lengths (Tensor): Lengths of the targets. 1-D Tensor of shape `(B,)`. blank_id (int, optional): The index of blank symbol in CTC emission. (Default: 0) ignore_id (int, optional): The index of ignore symbol in CTC emission. (Default: -1) )devicer)dim) start_dimNFz-inf)r dtyper)min)r index)sizetorcharanger catstack full_likeflatten as_tensorexpandtensorfloatrfullzerosrangewheremaxgatherlongargmaxclamp)rrrrrr batch_sizeinput_time_size_ bsz_indices_t_a_r_g_e_t_s_ diff_labelsneg_inf padding_numpadded_t best_score backpointerstprevprev_max_value prev_max_idxl1l2pathtarget_indices alignmentsr:a/home/ubuntu/.local/lib/python3.10/site-packages/funasr/models/sense_voice/utils/ctc_alignment.pyctc_forced_alignsP  &"*B,&($r<)rr)rTensorintr<r:r:r:r;s$