(self, linear_width)
| 646 | input_dims = output_dims = 1 |
| 647 | |
| 648 | def __init__(self, linear_width): |
| 649 | super().__init__() |
| 650 | if linear_width <= 0.0: |
| 651 | raise ValueError("Scale parameter 'linear_width' " + |
| 652 | "must be strictly positive") |
| 653 | self.linear_width = linear_width |
| 654 | |
| 655 | def transform_non_affine(self, values): |
| 656 | return self.linear_width * np.arcsinh(values / self.linear_width) |