tf.squeeze()函數的作用是從tensor中刪除所有大小(szie)是1的維度。
給定丈量輸入, 此操作返回的是相同類型的張量, 並刪除所有尺寸為1的維度。如果不想刪除所有尺寸為1的維度,
可以通過指定squeeze_dims來刪除特定維度。
下面通過例子來理解:
# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
shape(squeeze(t)) ==> [2, 3] # 可見, 把shape為1的維度都刪除了。
或者刪除特定的維度:
# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
shape(squeeze(t, [2, 4])) ==> [1, 2, 3, 1]