def erf_approx(self, x):
exp = -x * x * (4 / math.pi + self.a_for_erf * x * x) / (1 + self.a_for_erf * x * x)
return torch.sign(x) * torch.sqrt(1 - torch.exp(exp))
微信
公众号
扫码关注公众号