From 48a62b7ca3bf3330b541ab8d18a3b0ede2237c8c Mon Sep 17 00:00:00 2001 From: Samuel Caldas Date: Thu, 21 May 2020 23:04:36 -0300 Subject: [PATCH] Exposing softmax activation --- src/TensorFlowNET.Core/APIs/tf.nn.cs | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/TensorFlowNET.Core/APIs/tf.nn.cs b/src/TensorFlowNET.Core/APIs/tf.nn.cs index c8ce62f9..19afce1d 100644 --- a/src/TensorFlowNET.Core/APIs/tf.nn.cs +++ b/src/TensorFlowNET.Core/APIs/tf.nn.cs @@ -116,6 +116,8 @@ namespace Tensorflow public IActivation relu() => new relu(); public IActivation swish() => new swish(); public IActivation tanh() => new tanh(); + + public IActivation softmax() => new softmax(); public Tensor tanh(Tensor x, string name = null) => gen_nn_ops.tanh(x, name);