AttributeError: 'module' object has no attribute 'affine_grid'

I have updated my pytorch to the master branch but am unable to understand why I am getting the following error AttributeError: 'module' object has no attribute 'affine_grid'
I also did import import torch.nn.functional as F but am still facing the error.

Edit 1 :
I have also checked the functional.py file and did find the affine_grid function. But still am unable to resolve this error.