tf.layers.Model和tf.layers.Layer

Question

老师在实验wide&deep模型中继承了tf.layers.Model，在本节自定义层的时候又使用了tf.layers.Layer，这两个api有什么区别呢？在这里可以互换么？

Answer 1

正十七 2019-12-22 20:18:11

两个的API不一样，model的话，你可以直接compile，fit，evaluate以及save。但是Layer就只是一层，还需要用Model再封装一次才能这样。

https://www.tensorflow.org/api_docs/python/tf/keras/Model?version=stable

https://www.tensorflow.org/api_docs/python/tf/keras/layers/Layer?version=stable

• 正十七 #1

  Layer相当于更小的粒度，所以同一个功能基本上都可以用这两个API分别实现，但是不是简单的互换就能解决的。

• 提问者 qq_雨后天晴_0 #2

  非常感谢！