Apple Interview Question

Comment on different PyTorch code snippets and fill it in with the training loop code.