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

This commit is contained in:
Hoël Bagard
2024-01-31 12:55:24 +09:00
committed by GitHub
parent 8018337109
commit 68ae493297

View File

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