Can we get the energy value of AttentionSeq2Seq？

[bityangke]

For visualization， the energy value of every time step is needed. I read the codes, it seems I need to change somewhere in the recurrentshop lib.