def _forward(self, x):
w = lu_reconstruct(lower_upper=self.lower_upper,
perm=self.permutation,
validate_args=self.validate_args)
return linear_operator_util.matmul_with_broadcast(
w, x[..., tf.newaxis])[..., 0]
def _forward(self, x):
lu, perm = self._broadcast_params()
w = lu_reconstruct(lower_upper=lu,
perm=perm,
validate_args=self.validate_args)
return tf.linalg.matvec(w, x)
def _forward(self, x):
w = lu_reconstruct(lower_upper=self.lower_upper,
perm=self.permutation,
validate_args=self.validate_args)
return tf.linalg.matvec(w, x)