torch.nn.CosineSimilarity --> IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)

Question

I have the following code and I am trying to extract a 2048 dimension vector from the pretrained resnet50 network.

import torch
import torchvision
import torc