o i@@sGdddeZdS)c@s(eZdZ d ddZddZd d Zd S) WPE fullcCs"||_||_||_||_||_dS)Ntapsdelay iterations psd_contextstatistics_mode)selfrrr r r r H/home/ubuntu/.local/lib/python3.10/site-packages/espnet/transform/wpe.py__init__s  z WPE.__init__cCs$dj|jj|j|j|j|j|jdS)Nzw{name}(taps={taps}, delay={delay}iterations={iterations}, psd_context={psd_context}, statistics_mode={statistics_mode}))namerrr r r )format __class____name__rrr r r )r r r r__repr__ sz WPE.__repr__cCs>ddlm}||d|j|j|j|j|jd}|dddS)zReturn enhanced :param np.ndarray xs: (Time, Channel, Frequency) :return: enhanced_xs :rtype: np.ndarray r)wpe)rrrr) nara_wpe.wper transposerrr r r )r xsrr r r__call__s z WPE.__call__N)rrrrr)r __module__ __qualname__rrrr r r rrs   rN)objectrr r r rs