o pi @sddlmZ ddefddZ    dded eed eed eed eedef d dZ dddZ    dded eed eed eed eedef ddZ ddefddZ    dded eed eed eed eedef ddZdS))ListreturncCs$d|d|||dd|S)aCompute expected number of frames after 1D convolution Parameters ---------- num_samples : int Number of samples in the input signal kernel_size : int Kernel size stride : int Stride padding : int Padding dilation : int Dilation Returns ------- num_frames : int Number of frames in the output signal Source ------ https://pytorch.org/docs/stable/generated/torch.nn.Conv1d.html#torch.nn.Conv1d r) num_samples kernel_sizestridepaddingdilationrrX/home/ubuntu/.local/lib/python3.10/site-packages/pyannote/audio/utils/receptive_field.pyconv1d_num_framess$rNrr r r r c Cs6|}t||||D]\}}}} t||||| d}q |S)N)r r r r )zipr) rr r r r num_frameskspdrrr multi_conv_num_frames8s  rcCs(d|d|}||d|d|S)aeCompute size of receptive field Parameters ---------- num_frames : int, optional Number of frames in the output signal kernel_size : int Kernel size stride : int Stride padding : int Padding dilation : int Dilation Returns ------- size : int Receptive field size rrr)rr r r r effective_kernel_sizerrr conv1d_receptive_field_sizeHrrc C>|}ttt||||D]\}}}} t||||| d}q |S)N)rr r r r )reversedlistrr) rr r r r receptive_field_sizerrrrrrr multi_conv_receptive_field_sizeds"rcCs(d|d|}|||dd|S)aKCompute center of receptive field Parameters ---------- frame : int Frame index kernel_size : int Kernel size stride : int Stride padding : int Padding dilation : int Dilation Returns ------- center : int Index of receptive field center rrr)framer r r r rrrr conv1d_receptive_field_centerxrrrc Cr)N)rr r r r )rrrr) rr r r r receptive_field_centerrrrrrrr !multi_conv_receptive_field_centers"r!)rrrr)NNNN)rrrrr)rrrrr) typingrintrrrrrr!rrrr s|