I see…Thank you for the clarification
do you think there is a way to reshape inputs to compute the similarity of each channel (lets say our batchsize is always 1) and get an out of
torch.Size([2, 10, 10])?