def _inverse(self, y): lu, perm = self._broadcast_params() return lu_solve( lower_upper=lu, perm=perm, rhs=y[..., tf.newaxis], validate_args=self.validate_args)[..., 0]
def _inverse(self, y): return lu_solve(lower_upper=self.lower_upper, perm=self.permutation, rhs=y[..., tf.newaxis], validate_args=self.validate_args)[..., 0]