tf.squeeze

从张量的形状中删除大小为 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,)

文档链接

总结

以上是生活随笔为你收集整理的tf.squeeze的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。