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函数确实不需要了。之前应该是在测试版的原因才需要这个函数。
00
相似问题