如何实现张量流中的多级铰链损失

csz*_*rot 5 neural-network tensorflow

我想在张量流中实现多级铰链损耗.配方如下:

多级铰链损失功能

我发现当预测正确时很难得到第二个最大预测概率.我尝试使用tf.nn.top_k来计算它,但遗憾的是tf.nn.top_k没有实现渐变操作.那么我该如何实现呢?

Yar*_*tov 3

top_k有渐变,在0.8版本中添加在这里