【tensorflow】: 理解tf.squeeze()

理解tf.squeeze()

squeeze(
    input,
    axis=None,
    name=None,
    squeeze_dims=None
)

該函數返回一個張量,這個張量是將原始input中所有維度爲1的那些維都刪掉的結果

轉自:https://www.jianshu.com/p/a21c0bc10a38

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章