Julia - 如何使用泄漏的 relu？

Question

如何在 Julia 中使用leaky relu 函数？

以下命令不起作用：

nn.leaky_relu(features, alpha) 

在python中，有this function:

tf.nn.leaky_relu(
    features,
    alpha=0.2,
    name=None
)

谢谢

编辑

在@phg 的回答下，我将这个函数用于 Julia：

function leaky_relu(features, alpha)
    features = convert(TensorFlow.Tensor{Any}, features)
    alpha = convert(TensorFlow.Tensor{Any}, alpha)
    return max(alpha.*features, features)
end

Answer 1

我认为TensorFlow.jl 只复制 C++ API，而不是 Python API。而leaky_relu is defined 仅在 Python 中。但是实现应该很容易从原始版本转换为 Julia：

def leaky_relu(features, alpha=0.2, name=None):
    with ops.name_scope(name, "LeakyRelu", [features, alpha]):
        features = ops.convert_to_tensor(features, name="features")
        alpha = ops.convert_to_tensor(alpha, name="alpha")
        return math_ops.maximum(alpha * features, features)

（我没有安装任何库，所以我无法测试任何东西。）