diff options
| author |  Yong Tang <yong.tang.github@outlook.com> | 2018-06-02 22:13:21 +0000 |
|---|---|---|
| committer | Yong Tang <yong.tang.github@outlook.com> | 2018-06-02 22:14:06 +0000 |
| commit | b7150cffc5e36fe736e648c624cfb8b0cb411f1f (patch) | |
| tree | cea6b03109d44a2f8088697bebfe5059582777a9 /tensorflow/contrib/constrained_optimization | |
| parent | 13ceff2d4096554f195a3c865c1391500e172485 (diff) | |
Update keep_dims for reduce_max
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/contrib/constrained_optimization')
| -rw-r--r-- | tensorflow/contrib/constrained_optimization/python/swap_regret_optimizer.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/constrained_optimization/python/swap_regret_optimizer.py b/tensorflow/contrib/constrained_optimization/python/swap_regret_optimizer.py index 91b2486393..3791dae8d7 100644 --- a/tensorflow/contrib/constrained_optimization/python/swap_regret_optimizer.py +++ b/tensorflow/contrib/constrained_optimization/python/swap_regret_optimizer.py @@ -206,10 +206,10 @@ def _project_log_stochastic_matrix_wrt_kl_divergence(log_matrix): # For numerical reasons, make sure that the largest matrix element is zero # before exponentiating. - log_matrix -= standard_ops.reduce_max(log_matrix, axis=0, keep_dims=True) + log_matrix -= standard_ops.reduce_max(log_matrix, axis=0, keepdims=True) log_matrix -= standard_ops.log( standard_ops.reduce_sum( - standard_ops.exp(log_matrix), axis=0, keep_dims=True)) + standard_ops.exp(log_matrix), axis=0, keepdims=True)) return log_matrix |