r""" :return: log-probability of the parameter value under the prior :rtype: torch.Tensor
(self, x)
| 27 | return self._transform(x) if self._transform is not None else x |
| 28 | |
| 29 | def log_prob(self, x): |
| 30 | r""" |
| 31 | :return: log-probability of the parameter value under the prior |
| 32 | :rtype: torch.Tensor |
| 33 | """ |
| 34 | return super().log_prob(self.transform(x)) |
| 35 | |
| 36 | def load_state_dict(self, state_dict: Mapping[str, Any], *args, **kwargs): |
| 37 | Module.load_state_dict(self, state_dict, *args, **kwargs) |