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