Check this awesome PyTorch basic functions :