Pytorch's Identity() implementation/alternative in Fluxml?

thanks