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

精闢的一句話就是:axis等於哪一維,那一維被壓縮爲1(也就是不顯示了。) tensorflow的squeeze()函數則是默認壓縮維度爲1的那一維。也可通過squeeze_dim=[]來指定壓縮的具體維度。
相關文章
相關標籤/搜索