Searched refs:log_matrix (Results 1 – 1 of 1) sorted by relevance
199 def _project_log_stochastic_matrix_wrt_kl_divergence(log_matrix): argument213 log_matrix = log_matrix - standard_ops.reduce_max(214 log_matrix, axis=0, keepdims=True)215 log_matrix = log_matrix - standard_ops.log(217 standard_ops.exp(log_matrix), axis=0, keepdims=True))218 return log_matrix