def _batch_matmul(self, x, transpose_x=False): if transpose_x: x = array_ops.batch_matrix_transpose(x) diag_mat = array_ops.expand_dims(self._diag, -1) return math_ops.square(diag_mat) * x
def _batch_sqrt_matmul(self, x, transpose_x=False): if transpose_x: x = array_ops.batch_matrix_transpose(x) diag_mat = array_ops.expand_dims(self._diag, -1) return diag_mat * x
def _batch_matmul(self, x, transpose_x=False): if transpose_x: x = array_ops.batch_matrix_transpose(x) self._check_x(x) return x
def assert_symmetric(matrix): matrix_t = array_ops.batch_matrix_transpose(matrix) return control_flow_ops.with_dependencies( [check_ops.assert_equal(matrix, matrix_t)], matrix)