ops.square

mattdangerw · Nov 27, 2023 · 73b40fd · 73b40fd
1 parent c5ad052
commit 73b40fd
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/examples/nlp/ipynb/sentence_embeddings_with_sbert.ipynb b/examples/nlp/ipynb/sentence_embeddings_with_sbert.ipynb
@@ -576,8 +576,8 @@
     "        ep = encoder(positive)\n",
     "        en = encoder(negative)\n",
     "\n",
-    "        positive_dist = keras.ops.sum(keras.ops.power(ea - ep, 2), axis=1)\n",
-    "        negative_dist = keras.ops.sum(keras.ops.power(ea - en, 2), axis=1)\n",
+    "        positive_dist = keras.ops.sum(keras.ops.square(ea - ep), axis=1)\n",
+    "        negative_dist = keras.ops.sum(keras.ops.square(ea - en), axis=1)\n",
     "\n",
     "        positive_dist = keras.ops.sqrt(positive_dist)\n",
     "        negative_dist = keras.ops.sqrt(negative_dist)\n",

diff --git a/examples/nlp/md/sentence_embeddings_with_sbert.md b/examples/nlp/md/sentence_embeddings_with_sbert.md
@@ -565,8 +565,8 @@ class TripletSiamese(keras.Model):
         ep = encoder(positive)
         en = encoder(negative)
 
-        positive_dist = keras.ops.sum(keras.ops.power(ea - ep, 2), axis=1)
-        negative_dist = keras.ops.sum(keras.ops.power(ea - en, 2), axis=1)
+        positive_dist = keras.ops.sum(keras.ops.square(ea - ep), axis=1)
+        negative_dist = keras.ops.sum(keras.ops.square(ea - en), axis=1)
 
         positive_dist = keras.ops.sqrt(positive_dist)
         negative_dist = keras.ops.sqrt(negative_dist)

diff --git a/examples/nlp/sentence_embeddings_with_sbert.py b/examples/nlp/sentence_embeddings_with_sbert.py
@@ -382,8 +382,8 @@ def __init__(self, encoder, **kwargs):
         ep = encoder(positive)
         en = encoder(negative)
 
-        positive_dist = keras.ops.sum(keras.ops.power(ea - ep, 2), axis=1)
-        negative_dist = keras.ops.sum(keras.ops.power(ea - en, 2), axis=1)
+        positive_dist = keras.ops.sum(keras.ops.square(ea - ep), axis=1)
+        negative_dist = keras.ops.sum(keras.ops.square(ea - en), axis=1)
 
         positive_dist = keras.ops.sqrt(positive_dist)
         negative_dist = keras.ops.sqrt(negative_dist)