python - 没有 tf.keras.backend 的 lambda 层函数定义（Python Keras 包）

Question

tf.keras.layers.Lambda文档解释了如何在 lambda 层中定义函数。该文档提供以下功能作为示例，

def antirectifier(x):

    x -= K.mean(x, axis=1, keepdims=True)
    x = K.l2_normalize(x, axis=1)

    pos = K.relu(x)
    neg = K.relu(-x)

    return K.concatenate([pos, neg], axis=1)

model.add(Lambda(antirectifier))

但据此，tf.keras.backend必须用于对输入的 Tensor 对象进行操作。

有什么方法可以使用默认的 python 包和用户定义的函数来定义 lambda 函数的步骤。

如果可能，请提供一些示例。

标签： pythonpython-3.xtensorflowkeras

Answer 1

如果你不使用import tensorflow它的功能，绝对没有问题。

代码是完美的，就是这样。

只是import keras.backend as K

例子rounded = K.round(x)

---

这是 Keras 独立文档：https ://keras.io/layers/core/#lambda