1. 程式人生 > >python np.mean(axis) tf.squeeze(x)

python np.mean(axis) tf.squeeze(x)

精闢的一句話就是:axis等於哪一維,那一維被壓縮為1(也就是不顯示了。)

tensorflow的squeeze()函式則是預設壓縮維度為1的那一維。也可通過squeeze_dim=[]來指定壓縮的具體維度。