summaryrefslogtreecommitdiffstats
path: root/model-integration
diff options
context:
space:
mode:
authorLester Solbakken <lesters@oath.com>2019-06-20 10:53:05 +0200
committerLester Solbakken <lesters@oath.com>2019-06-20 10:53:05 +0200
commitc286fdc8e628b9141a54ccbf76c46ed36f498e34 (patch)
treed494e22a35606f83872fca34d56fc8d2eb013d6b /model-integration
parent6ad990b0309d673a279f9a1308f6029d028c11b7 (diff)
Add javadoc to softmax
Diffstat (limited to 'model-integration')
-rw-r--r--model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Softmax.java5
1 files changed, 5 insertions, 0 deletions
diff --git a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Softmax.java b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Softmax.java
index 8db237d15e4..cdacbe1656a 100644
--- a/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Softmax.java
+++ b/model-integration/src/main/java/ai/vespa/rankingexpression/importer/operations/Softmax.java
@@ -6,6 +6,11 @@ import com.yahoo.tensor.functions.TensorFunction;
import java.util.List;
+/**
+ * Convert imported 'softmax' operation to the Vespa softmax ranking function.
+ *
+ * @author lesters
+ */
public class Softmax extends IntermediateOperation {
public Softmax(String modelName, String nodeName, List<IntermediateOperation> inputs) {