Чем отличается функция unsqueeze от squeeze в PyTorch?

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

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

Функция unsqueeze вставляет новое измерение размера 1 в указанную позицию входного тензора. www.slingacademy.com www.geeksforgeeks.org Например, если у входного тензора форма (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). www.slingacademy.com