Incompatible with the new version of pytorch

[ultraman-blazar]

I think it's related to torch.nn.function, and this one has been replaced by torch.F.function so that cause the incompatible problems.