def _get_arctanh_x(self, x):
# Carlini's original implementation uses a slightly different formula because
# the image space is [-0.5, 0.5] instead of [clip_min, clip_max]
result = clamp((x - self.clip_min) / (self.clip_max - self.clip_min),
min=0.,
max=1.) * 2 - 1
return torch_arctanh(result * ONE_MINUS_EPS)
def _get_arctanh_x(self, x):
result = clamp((x - self.clip_min) / (self.clip_max - self.clip_min),
min=self.clip_min,
max=self.clip_max) * 2 - 1
return torch_arctanh(result * ONE_MINUS_EPS)
