嗨，我正在尝试计算从模型中提取的嵌入向量之间的距离。我正在研究图像检索问题。
当我使用keras Path: $['map']['photo']
Value: map photo
Path: $['person']['data']['photos']['photo']
Value: photo Base64
Path: $['person']['data']['photos']['passportPhoto']
Value: passport photo Base64
函数时，即使指定了ImageDataGenerator，模型的输出形状也例如是batch_size
但是当我使用pytorch时，如果将(number of query images or reference images, num_classes)与DataLoader一起使用，例如32，则模型的输出大小将变为batch_size。最后一层是线性层。
我需要像(32, num_classes)那样做形状，就像我对角膜所做的那样，因为我需要每个都嵌入矢量来计算距离。
有什么建议吗？谢谢