add sqrt func for ndarray

scicloj · Oct 5, 2021 · b544583 · b544583
1 parent 07fa04d
commit b544583
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/clj_djl/ndarray.clj b/src/clj_djl/ndarray.clj
@@ -702,3 +702,7 @@
  "Returns the inverse trigonometric cosine of this NDArray element-wise."
  [ndarray]
  (.acos ndarray))
+
+(defn sqrt
+ [ndarray]
+ (.sqrt ndarray))