softmax - 如何在chainer中实现加权softmax

Question

我正在重新制作一篇论文https://arxiv.org/abs/1711.11575：它有一个公式：

但是我搜索了chainer，它只有F.softmax，但它不能增加权重。如何重新实现该公式？

Answer 1

如果要添加 term w_G^{mn}，我猜在应用通常的 softmax 之前向每个输出添加值 log(w_G^{mn}) 应该具有相同的效果。