def forward(self, k, m, x, mean, std):
ks = Ft(x * std + mean)
k = k[:, :1] + 1j * k[:, 1:]
k = self.al(Ft(FtH(k)))
x = FtH((1 - m) * ks + m * (ks + self.ll * k) / (1 + self.ll)).abs()
ks = Ft(x)
x = FtH((1 - m) * ks + m * (ks + self.ll * k) / (1 + self.ll)).abs()
return (x - mean) / (std + 1e-11)
def forward(self, k, m, x, mean, std):
ks = Ft(x * std + mean)
x = FtH((1 - m) * ks + m * k)
return (x - mean) / (std + 1e-11 + 1j * 1e-11)
def forward(self, k, m, x, mean, std):
ks = Ft(x * std + mean)
k = k[:, :1] + 1j * k[:, 1:]
x = FtH((1 - m) * ks + m * k).abs()
return (x - mean) / (std + 1e-11)