Add tensor argmax and argmin

1 files changed, 1 insertions, 7 deletions

diff --git a/vespajlib/src/main/java/com/yahoo/tensor/functions/Diag.java b/vespajlib/src/main/java/com/yahoo/tensor/functions/Diag.java
index 007baa4fa70..aae63ff846e 100644
--- a/vespajlib/src/main/java/com/yahoo/tensor/functions/Diag.java
+++ b/vespajlib/src/main/java/com/yahoo/tensor/functions/Diag.java
@@ -1,12 +1,6 @@
 package com.yahoo.tensor.functions;
 
 import com.yahoo.tensor.TensorType;
-import com.yahoo.tensor.functions.CompositeTensorFunction;
-import com.yahoo.tensor.functions.Generate;
-import com.yahoo.tensor.functions.PrimitiveTensorFunction;
-import com.yahoo.tensor.functions.ScalarFunctions;
-import com.yahoo.tensor.functions.TensorFunction;
-import com.yahoo.tensor.functions.ToStringContext;
 
 import java.util.Collections;
 import java.util.List;
@@ -26,7 +20,7 @@ public class Diag extends CompositeTensorFunction {
     
     public Diag(TensorType type) {
         this.type = type;
-        this.diagFunction = ScalarFunctions.equalArguments(dimensionNames().collect(Collectors.toList()));
+        this.diagFunction = ScalarFunctions.equal(dimensionNames().collect(Collectors.toList()));
     }
     
     @Override