(self, linear_width)
| 664 | input_dims = output_dims = 1 |
| 665 | |
| 666 | def __init__(self, linear_width): |
| 667 | super().__init__() |
| 668 | self.linear_width = linear_width |
| 669 | |
| 670 | def transform_non_affine(self, values): |
| 671 | return self.linear_width * np.sinh(values / self.linear_width) |