add tf.where

python · Jan 5, 2025 · 4e678f2 · 4e678f2
1 parent 89d1bcb
commit 4e678f2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/stubs/tensorflow/tensorflow/__init__.pyi b/stubs/tensorflow/tensorflow/__init__.pyi
@@ -412,4 +412,7 @@ def pad(
     name: str | None = None,
 ) -> Tensor: ...
 def shape(input: TensorCompatible, out_type: DTypeLike | None = None, name: str | None = None) -> Tensor: ...
+def where(
+    condition: TensorCompatible, x: TensorCompatible | None = None, y: TensorCompatible | None = None, name: str | None = None
+) -> Tensor: ...
 def __getattr__(name: str) -> Incomplete: ...