标签: tensorflow keras
我有一个自定义图层MyLayer,该图层继承自keras.layers.Layer。
MyLayer
keras.layers.Layer
在Call函数内部,我想获取输入来源的图层,即:
Call
class MyLayer(keras.layers.Layer): ... def call(self, inputs): layer = inputs.layer ...
在TensorFlow中可以实现吗?如果没有,我有什么办法可以从call函数中获取输入的内容?
call