squeeze()函数

squeeze()函数是用来去除张量(tensor)中维度为1的维度的函数。它可以将维度为1的维度压缩掉，从而减少张量的维度。例如，对于一个维度为(1, 3, 1, 2)的张量，使用squeeze()函数可以将其压缩为(3, 2)的张量。需要注意的是，squeeze()函数默认只压缩维度为1的维度，如果想要压缩其他维度，可以通过指定维度的参数来实现。