调用自定义keras层时获取张量层

时间:2020-10-13 11:01:25

标签: tensorflow keras

我有一个自定义图层MyLayer,该图层继承自keras.layers.Layer

Call函数内部,我想获取输入来源的图层,即:

class MyLayer(keras.layers.Layer):
   ...
   def call(self, inputs):
       layer = inputs.layer
   ...

在TensorFlow中可以实现吗?如果没有,我有什么办法可以从call函数中获取输入的内容?

0 个答案:

没有答案