diff --git a/stubs/tensorflow/tensorflow/math.pyi b/stubs/tensorflow/tensorflow/math.pyi index 5c300503e..d3b02bf7a 100644 --- a/stubs/tensorflow/tensorflow/math.pyi +++ b/stubs/tensorflow/tensorflow/math.pyi @@ -24,6 +24,10 @@ def abs(x: SparseTensor, name: str | None = None) -> SparseTensor: ... @overload def abs(x: RaggedTensor, name: str | None = None) -> RaggedTensor: ... @overload +def angle(input: TensorCompatible, name: str | None = None) -> Tensor: ... +@overload +def angle(input: RaggedTensor, name: str | None = None) -> RaggedTensor: ... +@overload def sin(x: TensorCompatible, name: str | None = None) -> Tensor: ... @overload def sin(x: RaggedTensor, name: str | None = None) -> RaggedTensor: ...