Skip to content

Commit bcf5ced

Browse files
authored
Update methods_comparison.py
1 parent fb527fd commit bcf5ced

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

dca/methods_comparison.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def make_norm_power(X, T_ent):
2626
"""Calculate the normalize power spectrum."""
2727
Y = F.unfold(X, kernel_size=[T_ent, 1], stride=T_ent)
2828
Y = torch.transpose(Y, 1, 2)
29-
Yf = torch.rfft(Y, 1, onesided=True)
29+
Yf = torch.fft.rfft(Y, 1, onesided=True)
3030
Yp = torch.mean(Yf[:, :, :, 0]**2 + Yf[:, :, :, 1]**2, dim=1)
3131
return Yp / torch.sum(Yp, dim=1, keepdim=True)
3232

0 commit comments

Comments
 (0)