Lambda层没有要传播的渐变

Question

我想添加一个Lamda层sum = Lambda(lambda x: K.sum(x, axis=1)(inputs)，但我不想传播渐变。我希望将其视为Input层。我该如何在Keras中做到这一点？类似于no_grad功能。

Answer 1

这应该有效

sum = Lambda(lambda x: K.sum(x, axis=1)(inputs)
sum_stop_grad = Lambda(lambda x: K.stop_gradient(x))(sum)
dense_1 = Dense(16)(sum_stop_grad)
...