From 824258e5d1d586905f5a65baf805ce61e1df64ca Mon Sep 17 00:00:00 2001 From: Robin Scheibler Date: Thu, 23 May 2024 13:03:22 +0900 Subject: [PATCH] black --- pyroomacoustics/bss/fastmnmf.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/pyroomacoustics/bss/fastmnmf.py b/pyroomacoustics/bss/fastmnmf.py index 6ba6b9d7..6caf833e 100644 --- a/pyroomacoustics/bss/fastmnmf.py +++ b/pyroomacoustics/bss/fastmnmf.py @@ -161,9 +161,7 @@ def separate(): Y_FTM = np.einsum("nft, nfm -> ftm", lambda_NFT, G_NFM) # update G_NFM (diagonal element of spatial covariance matrices) - numerator = np.einsum( - "nft, ftm -> nfm", lambda_NFT, Qx_power_FTM / (Y_FTM**2) - ) + numerator = np.einsum("nft, ftm -> nfm", lambda_NFT, Qx_power_FTM / (Y_FTM**2)) denominator = np.einsum("nft, ftm -> nfm", lambda_NFT, 1 / Y_FTM) + eps G_NFM *= np.sqrt(numerator / denominator) Y_FTM = np.einsum("nft, nfm -> ftm", lambda_NFT, G_NFM)