從張量的形狀中刪除大小為 1 的維度

import tensorflow as tfa=tf.constant([[[1,2,3,1]]])print(a.shape)a1=tf.squeeze(a)print(a1.shape) (1, 1, 4) (4,)

文檔鏈接

總結(jié)

以上是生活随笔為你收集整理的tf.squeeze的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。