Yang Yu

@thomashirtz I have the same problem. Have you solved it？

> Thank a lot for your patience. I have tried a trick and address it. The main idea is to control the index in order to using softmax function. We...