As @melgor said, you are still mixing Module (collection of autograd operations) and Function (a custom autograd option).
Read these:
Function: http://pytorch.org/tutorials/beginner/pytorch_with_examples.html#pytorch-defining-new-autograd-functions 31
Module: http://pytorch.org/tutorials/beginner/pytorch_with_examples.html#pytorch-custom-nn-modules 22