Gumbel_Softmax 概要

Gumble_Softmax 可以解决的问题 **场景:**对于一个分类任务,通常会使用softmax函数来将模型的输出值转换为概率的形式,并通过argmax函数取最大的概率值标签作为模型的预测标签。在分类任务中,argmax可以不参与反向传播过程(即直接通过softmax值和true_label计

你可能感兴趣的:(Gumbel_Softmax 概要)