torch.nn torch.nn If you really want to understand what this does lol https://docs.pytorch.org/tutorials/beginner/nn_tutorial.html#neural-net-from-scratch-without-torch-nn