diff --git a/src/TensorFlowNET.Core/APIs/tf.nn.cs b/src/TensorFlowNET.Core/APIs/tf.nn.cs index 7b9d0a2c..01789617 100644 --- a/src/TensorFlowNET.Core/APIs/tf.nn.cs +++ b/src/TensorFlowNET.Core/APIs/tf.nn.cs @@ -163,6 +163,9 @@ namespace Tensorflow => gen_nn_ops.local_response_normalization(input, depth_radius: depth_radius, bias: bias, alpha: alpha, beta: beta, name: name); + public Tensor leaky_relu(Tensor features, float alpha = 0.2f, string name = null) + => nn_ops.leaky_relu(features, alpha: alpha, name: name); + public rnn_cell_impl rnn_cell => new rnn_cell_impl(); public Tensor softmax(Tensor logits, int axis = -1, string name = null) diff --git a/src/TensorFlowNET.Core/Operations/NnOps/gen_nn_ops.cs b/src/TensorFlowNET.Core/Operations/NnOps/gen_nn_ops.cs index f1e40151..49d504ab 100644 --- a/src/TensorFlowNET.Core/Operations/NnOps/gen_nn_ops.cs +++ b/src/TensorFlowNET.Core/Operations/NnOps/gen_nn_ops.cs @@ -214,6 +214,17 @@ namespace Tensorflow.Operations }); return _op.outputs[0]; + } + + public static Tensor leaky_relu(Tensor features, float alpha = 0.2f, string name = null) + { + var _op = _op_def_lib._apply_op_helper("LeakyRelu", name: name, args: new + { + features, + alpha + }); + + return _op.output; } public static Tensor max_pool(Tensor input,