Persona:

I want to make a structure like the one shown above, but now it seems to be the following structure.


It is the structure you want.



 Persona:

How can I get Integer prediction value?


predict = output.data.max(1)[1]