Skip to content

Commit 48a62b7

Browse files
samuelcaldasOceania2018
authored andcommitted
Exposing softmax activation
1 parent c172126 commit 48a62b7

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/TensorFlowNET.Core/APIs/tf.nn.cs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,6 +116,8 @@ public Tensor embedding_lookup(Tensor @params,
116116
public IActivation relu() => new relu();
117117
public IActivation swish() => new swish();
118118
public IActivation tanh() => new tanh();
119+
120+
public IActivation softmax() => new softmax();
119121
public Tensor tanh(Tensor x, string name = null)
120122
=> gen_nn_ops.tanh(x, name);
121123

0 commit comments

Comments
 (0)