def mse_complex(x, y):
    diff = x - y
    return torch.mean(diff.real ** 2 + diff.imag ** 2)