如何测试 Keras 自定义层的正确性？

Question

创建具有训练权重的 Keras 自定义层后，如何测试代码的正确性？Keras的手册中似乎没有描述。

例如，要测试函数的预期行为，可以编写单元测试。我们如何为 Keras 自定义层执行此操作？

Answer 1

您仍然可以通过获取给定输入的自定义层的输出并根据手动计算的输出进行验证来执行单元测试之类的操作，

假设您的自定义图层Custom采用 (None, 3, 200) 作为输入形状并返回 (None, 3)

from keras.layers import Input
from keras.models import Model

inp = Input(shape=(3, 200))
out = Custom()(inp)
model = Model(inp, out)

output = model.predict(your_input)

output您可以使用已知输入的预期输出来验证层输出your_input。