I\'m going through cs231n exercises. Now I have a problem with it! In assignment 2 in PyTorch.ipynb (The link of the file : https://github.com/israfelsr/CS231n/blob/master/assig
