y_pred = tf.squeeze(y_pred, 1) 这句的作用是什么

来源：3-12 tf.GradientTape与tf.keras结合使用

刘鹏4

2021-07-28

        y_pred = tf.squeeze(y_pred, 1)  这句的作用是什么

写回答

1回答

正十七

2021-08-07

把size为1的维度给去掉。

在那个模型中，因为最后一层是size=1的dense layer, 所以得到的输出是y_pred.shape = [batch_size，1], 而y_batch.shape = [batch_size], 所以对应不上，需要把y_pred的axis=1处（也就是第二个维度）给干掉。

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

相似问题

回答 1

回答 1

回答 1

回答 1

pass

回答 1