Solved by V_Deamo in post #6 
              
              
                Firstly, I need to modify my answer, as far as I concerned that summary can not directly summary the model with 5 outputs in forward function. You can simply don’t use the default function forward for encoder. You can change the default forward to forward_any_name_you_want to compute the results, th…