Hi @Eta_C thanks for your reply. I am currently using pytorch 0.4.0 for my task so Identity function is not supported in this version i am supposing because of which it is unavailable. Is there any other alternative?