Commit 6c22bbdd authored by Yong Tang's avatar Yong Tang Committed by Jonathan Hseu
Browse files

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: default avatarYong Tang <yong.tang.github@outlook.com>
parent 0c2ca00e
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment