deep-learning - 如何测试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 。