Tensorflow 调用自定义keras层时获取张量层
我有我的自定义图层Tensorflow 调用自定义keras层时获取张量层,tensorflow,keras,Tensorflow,Keras,我有我的自定义图层MyLayer,它继承自keras.layers.layer 在调用函数中,我想获取输入来自的层,即: class MyLayer(keras.layers.Layer): ... def call(self, inputs): layer = inputs.layer ... 这可以在TensorFlow中实现吗?如果没有,我有没有办法从调用函数获取输入层
MyLayer
,它继承自keras.layers.layer
在调用
函数中,我想获取输入来自的层,即:
class MyLayer(keras.layers.Layer):
...
def call(self, inputs):
layer = inputs.layer
...
这可以在TensorFlow中实现吗?如果没有,我有没有办法从调用
函数获取输入层