1 . 网址: https://www.reddit.com/r/MachineLearning/comments/4e2get/argmax_differentiable/
2. tensorflow参看函数的反向传播
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/ops/math_grad.py#L542
3. argmax的反向传播:
(1)THE CONCRETE DISTRIBUTION: THE CONCRETE DISTRIBUTION:
A CONTINUOUS RELAXATION OF DISCRETE RANDOM VARIABLES。
(2)CATEGORICAL REPARAMETERIZATION WITH GUMBEL-SOFTMAX
argmax in DNN
最新推荐文章于 2024-12-07 13:51:17 发布