数据挖掘 - 继承另一个类的自定义张量流层 - 吾爱随笔录

数据挖掘喀拉斯张量流

2022-02-26 01:31:40

我正在尝试在 tensorflow 中创建自己的自定义层，如下所示：

class custom_LSTM(RNN):
    # implementation of class

其中 RNN 是另一个类。

但是，要在模型中使用此自定义层，它需要继承tf.keras.layers.Layer. 我正在使用tf.keras.

知道如何解决这个问题吗？

1个回答

虽然我不是 RNN 的出口，但鉴于 RNN 是从 Layer 继承的，以下应该可以工作：

from keras.layers import RNN
class custom_LSTM(RNN):
   # implementation of class
   ...

提供可以重现您的错误的最小代码段可能会很好。

其它你可能感兴趣的问题