Thanks for the reply. I’ll check this one too.
Beside, I found that nn::sequential also can be used for a same purpose, though it does not need a forward implementation, which can be a positive point and at a same time a negative point.
Thanks,
Afshin