hi @valhalla
Thank you very much for your reply. As a feedback, it would be good if it is mentioned in the documentation of .predict() for Seq2Seq. Or did I miss it?
Okay so you mean in that case I do not need to take to logist and argmax them? I directly get the results?
I will try your solution and let you know