Motion blur with PyTorch autograd