Привет. Я пытаюсь вычислить расстояние между векторами встраивания, которое извлекается из модели.Я работаю над проблемой поиска изображений.когда я работаю с функцией keras ImageDataGenerator, даже когда указан batch_size, выходная форма модели, например, (number of query images or reference images, num_classes)
, но когда я работаю с pytorch, при использовании DataLoader сbatch_size скажем, 32, выходной размер модели становится (32, num_classes).Последний слой является линейным слоем.
Мне нужна форма (number of query images or reference images, num_classes), чтобы сделать то же самое, что я делал с кератами, потому что мне нужно встраивать векторы каждый для вычисления расстояния.Любой совет, пожалуйста?спасибо