Pytorch can be considered as an enhanced NumPy, that uses the power of GPU's. It is also packed with different modules that make it a good platform for deep learning research.

Pytorch works with Tensors - that are similar to the ndarrays in NumPy. But, Tensors use the power of GPU's.

For working with PyTorch, we need to start with importing the pytorch module.

import torch