add tf.where

Author: hoel-bagard

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: ...