摘要: Gumble_Softmax 可以解决的问题 **场景:**对于一个分类任务,通常会使用softmax函数来将模型的输出值转换为概率的形式,并通过argmax函数取最大的概率值标签作为模型的预测标签。在分类任务中,argmax可以不参与反向传播过程(即直接通过softmax值和true_label计 阅读全文
posted @ 2022-08-09 10:30 jacknie23 阅读(3839) 评论(1) 推荐(0) 编辑