Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def _apply(self, fn):
"""
Mimics the behavior of the function _apply() of a nn.Module()
"""
_apply_psi(self.psi1_f, fn)
_apply_psi(self.psi2_f, fn)
_apply_phi(self.phi_f, fn)
return self