o
    ¡¿¯iÛ  ã                   @   sJ   d dl mZmZ d dlmZ d dlmZ d dlZG dd„ dejj	eƒZ
dS )é    )ÚABCÚabstractmethod)ÚOrderedDict)ÚTupleNc                   @   sF   e Zd Zeedefdd„ƒƒZedeeej	 ej	e
f fdd„ƒZdS )ÚAbsMaskÚreturnc                 C   ó   t ‚©N©ÚNotImplementedError)Úself© r   úP/home/ubuntu/.local/lib/python3.10/site-packages/espnet2/diar/layers/abs_mask.pyÚmax_num_spk	   s   zAbsMask.max_num_spkc                 C   r   r	   r
   )r   ÚinputÚilensÚbottleneck_featÚnum_spkr   r   r   Úforward   s   	zAbsMask.forwardN)Ú__name__Ú
__module__Ú__qualname__Úpropertyr   Úintr   r   ÚtorchÚTensorr   r   r   r   r   r   r      s    úr   )Úabcr   r   Úcollectionsr   Útypingr   r   ÚnnÚModuler   r   r   r   r   Ú<module>   s
    