Thank you, I’ll check that out. I guess that’s what I actually need.
Just some other questions out of curiosity:

What does functional mean in this context?
Why isn’t nn.Module functional?
Will this problem be addressed in a future release?