projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 0c2ca00) | patch
Fix warnings in tf.distributions.Categorical (#18468)
authorYong Tang <yong.tang.github@outlook.com>
Fri, 13 Apr 2018 18:47:02 +0000 (11:47 -0700)
committerJonathan Hseu <vomjom@vomjom.net>
Fri, 13 Apr 2018 18:47:02 +0000 (11:47 -0700)
commit6c22bbdda41d839cb9e1f7803533c571596ea4ee
treeec73d204a3c425d615bbef94e869d4ccb4ab4305tree | snapshot
parent0c2ca00e1082ab2692af68af183083e41393f6c4commit | diff
Fix warnings in tf.distributions.Categorical (#18468)

In tf.distributions.Categorical dimension was used with argmax.
As dimension has been deprecated this generates a warning.
This fix fixes the warning by changing to axis.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
tensorflow/python/ops/distributions/categorical.py diff | blob | history
Domain: Machine Learning / ML Framework;