tf.squeeze

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

用于压缩张量中为1的轴,去除张量中形状为1的轴。

例:

import tensorflow as tf

raw = tf.Variable(tf.random_normal(shape=(1, 3, 2)))
squeezed = tf.squeeze(raw)

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(raw.shape)
    print(sess.run(squeezed).shape)


Output:
-------------
(1, 3, 2)
(3, 2)
-------------

转载地址:https://blog.csdn.net/loseinvain/article/details/78994695

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券