Skip to content

Commit

Permalink
black
Browse files Browse the repository at this point in the history
  • Loading branch information
fakufaku committed May 23, 2024
1 parent 3429338 commit 824258e
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions pyroomacoustics/bss/fastmnmf.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 824258e

Please sign in to comment.