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]