@vadimkantorov is right. This will be improved before 0.4 is released once Variable and Tensor are merged.  The master branch is in an intermediate state where autograd Functions have to return Variables and not Tensors.