Understanding dimensions in PyTorch
When I started doing some basic operations with PyTorch tensors like summation, it looked easy and pretty straightforward for one-dimensional tensors: However, once I started to play around with 2D and 3D tensors and to sum over rows and columns, I got confused mostly about the second parameter of…