model2.build函数的必要性

来源:7-8 文本生成实战之采样生成文本

wxz123

2019-12-28

老师对于
model2 = build_model(vocab_size, embedding_dim, rnn_units, batch_size = 1)
配合之前的
keras.layers.Embedding(vocab_size, embedding_dim,batch_input_shape = [batch_size, None]),
此处不已经将输入固定为[1, None]了吗,为什么还要设置model2.build(tf.TensorShape([1, None]))这句代码来设置输入的大小呢

写回答

1回答

正十七

2020-01-04

同学你好,经过验证,在2.0正式版上build函数确实不需要了。之前应该是在测试版的原因才需要这个函数。

0
0

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

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

1849 学习 · 896 问题

查看课程