Tensores Pytorch - Uso de squeeze e unsqueeze

Oii Geraldo, tudo bem?? Espero que sim!

Desculpa a demora em te dar um retorno por aqui.

As funções squeeze e unsqueeze são bem complicadas mesmo, mas segundo a documentação do Pytorch, o método squeeze retorna um tensor com todas as dimensões de input com o tamanho de 1 removido e o método unsqueeze retorna um novo tensor com uma dimensão inserida na posição especificada, mas isso ainda não esclarece o que é realmente cada método.

De forma mais simples, o método unsqueeze adiciona uma dimensão superficial ao tensor (na dimensão que for especificada) enquanto o método squeeze remove uma dimensão do tensor, ou seja, se ele for 3D antes do squeeze, depois de aplicado vira 2D.

Na imagem podemos ver uma representação de squeeze e unsqueeze para esse exemplo da matriz 2D.

A interpretação que podemos tirar é que independente do tensor 3D, se você aplicar o squeeze, terá o mesmo resultado, sem precisar especificar a dimensão. No método unsqueeze é um pouco diferente, já que é necessário passar o parâmetro da dimensão que ele vai criar, como o exemplo da imagem para o lado ou para trás (linha ou coluna).

Esses métodos são utilizados para comprimir e descomprimir tensores e depende da necessidade do seu problema, já que esses tensores vão ter a mesma informação, apenas índices diferentes para serem acessados. Então digamos que você esteja trabalhando com diversos tensores de tamanhos diferentes e queira deixar eles com o mesmo tamanho, é necessário aplicar estas técnicas.

Espero ter ajudado, mas qualquer dúvida pode entrar em contato comigo, ok?

Bons estudos e um ótimo ano novo ^^