Which PyTorch version are you using? Mine might be a bit too old (1.0.0.dev20190104).
1.0.0.dev20190104