o ߥi*@sXddlZddlZddlZddlmZddZddZddZGdd d ej ej d Z dS) NcCs`td||}|durdS||d}td|}|dur"dS||d|dfS)Nz^\s*z^\s*(-?\d+\.?\d*e?-?\d*?)r)rematchendgroups)instrtoken first_tokenlrr a/home/ubuntu/.local/lib/python3.10/site-packages/modelscope/models/audio/aec/layers/layer_base.pyexpect_token_number s r cCsx|dd}|d|}g}||d|dD]}tj|tjdd}|jdkr.||q||ddt|fS)N[r]  )dtypesep)findsplitnp fromstringfloat32sizeappendarray)rpos2pos3matstttmp_matr r r expect_kaldi_matrixs    r!cCs<tjtjtjddt|}|dd}|dd}d|S)z} function that transform as str numpy mat to standard kaldi str matrix :param np_mat: numpy mat :return: str T) threshold linewidthsuppressr rz[ %s ] )rset_printoptionsinfnanstrreplace)np_matout_strr r r to_kaldi_matrix s   r-cs*eZdZfddZejddZZS) LayerBasecstt|dSN)superr.__init__self __class__r r r1/szLayerBase.__init__cCsdSr/r r2r r r to_kaldi_nnet2szLayerBase.to_kaldi_nnet)__name__ __module__ __qualname__r1abcabstractmethodr6 __classcell__r r r4r r.-s r.) metaclass) r:rnumpyrtorch.nnnnr r!r-ModuleABCMetar.r r r r s