Как я могу сохранить векторы функций для каждого изображения в моей папке в Pytorch

Question

Я обучил свою модель следующим образом: теперь я хочу снова вставить свои обучающие данные в эту обученную модель и извлечь вектор признаков для каждого изображения, чтобы его размер был "встроенным размером" в моем коде. И затем я хочу сохранить все эти векторы функций на своем жестком диске. С другой стороны, у меня много папок с изображениями, и я хочу сохранить векторы изображений так же, как эти папки с изображениями и с теми же именами.

Пожалуйста, помогите мне, как сделать это с помощью кода?

class TripletNetwork(nn.Module):
    def __init__(self, cnn, embed_size):
        super(TripletNetwork, self).__init__()
        num_features = cnn.fc.in_features
        cnn.fc = nn.Sequential(
            nn.Linear(num_features, 512),
            nn.BatchNorm1d(512),
            nn.ReLU(),           
            nn.Linear(512, embed_size))
        self.cnn = cnn

    def forward(self, anc, pos, neg):
        f_anc = self.cnn(anc)
        f_pos = self.cnn(pos)
        f_neg = self.cnn(neg)
        return f_anc, f_pos, f_neg