Функции unsqueeze и squeeze в PyTorch отличаются тем, что каждая из них выполняет противоположные действия. 14

Функция squeeze удаляет все измерения размера 1 у тензора. 13 Например, если у входного тензора форма (A x 1 x B x C x 1 x D), то функция squeeze вернёт тензор формы (A x B x C x D) после удаления измерений размера 1. 1

Функция unsqueeze вставляет новое измерение размера 1 в указанную позицию входного тензора. 13 Например, если у входного тензора форма (A x B x C), то функция unsqueeze может добавить новое измерение в любой позиции: (A x 1 x B x C), (A x B x 1 x C) или (A x B x C x 1). 1