tensorflow: add tensorflow.math.reduce_variance (#11328)

python · Jan 31, 2024 · 68ae493 · 68ae493
1 parent 8018337
commit 68ae493
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/stubs/tensorflow/tensorflow/math.pyi b/stubs/tensorflow/tensorflow/math.pyi
@@ -252,6 +252,12 @@ def reduce_std(
     keepdims: bool = False,
     name: str | None = None,
 ) -> Tensor: ...
+def reduce_variance(
+    input_tensor: _TensorCompatible | RaggedTensor,
+    axis: _TensorCompatible | None = None,
+    keepdims: bool = False,
+    name: str | None = None,
+) -> Tensor: ...
 def argmax(
     input: _TensorCompatible, axis: _TensorCompatible | None = None, output_type: _DTypeLike = ..., name: str | None = None
 ) -> Tensor: ...