Simplifying tensor operations with einops library

Let's look at an example of reshaping a tensor with einops. Assume we have a 4D shape tensor (batch_size, height, width, channels) representing a batch of RGB photos. We wish to reshape the tensor to have the shape (batch_size, channels, height * wid...