如何在 chainer 中实现 weighted-softmax

How to implement weighted-softmax in chainer

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

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

如果你想添加术语 w_G^{mn},我想在应用通常的 softmax 之前对每个输出添加 "adding a value log(w_G^{mn})" 应该有相同的效果。