Browse Source

Exposing softmax activation

tags/v0.20
Samuel Caldas Haiping 5 years ago
parent
commit
48a62b7ca3
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      src/TensorFlowNET.Core/APIs/tf.nn.cs

+ 2
- 0
src/TensorFlowNET.Core/APIs/tf.nn.cs View File

@@ -116,6 +116,8 @@ namespace Tensorflow
public IActivation relu() => new relu(); public IActivation relu() => new relu();
public IActivation swish() => new swish(); public IActivation swish() => new swish();
public IActivation tanh() => new tanh(); public IActivation tanh() => new tanh();

public IActivation softmax() => new softmax();
public Tensor tanh(Tensor x, string name = null) public Tensor tanh(Tensor x, string name = null)
=> gen_nn_ops.tanh(x, name); => gen_nn_ops.tanh(x, name);




Loading…
Cancel
Save