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)