!47055 fix reciprocal when input is complex

Merge pull request !47055 from 冯一航/fix_reciprocal_input_complex
This commit is contained in:
i-robot 2022-12-22 02:13:53 +00:00 committed by Gitee
commit 8369496cde
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
3 changed files with 4 additions and 8 deletions

View File

@ -12,8 +12,6 @@ mindspore.ops.sign
1 & \text{x} > 0
\end{cases}
如果输入为nan则返回0。
参数:
- **x** (Tensor) - 输入Tensor。

View File

@ -1689,8 +1689,6 @@ def sign(x):
\frac{{\text{x}_i}}{|{\text{x}_i}|} & \text{otherwise}
\end{cases}
0 is returned for nan inputs.
Args:
x (Tensor): Input Tensor.
@ -1772,7 +1770,7 @@ def sgn(x):
TypeError: If input is not a Tensor.
Supported Platforms:
``Ascend`` ``GPU`` ``CPU``
``GPU`` ``CPU``
Examples:
>>> import mindspore as ms
@ -3966,7 +3964,7 @@ def nan_to_num(x, nan=0.0, posinf=None, neginf=None):
TypeError: If dtype of `x` is not float16 or float32.
Supported Platforms:
``Ascend`` ``CPU``
``CPU``
Examples:
>>> x = Tensor(np.array([float('nan'), float('inf'), -float('inf'), 3.14]), mindspore.float32)
@ -4536,7 +4534,7 @@ def reciprocal(x):
>>> print(output)
[1. 0.5 0.25]
"""
if not ops.is_floating_point(x):
if not is_complex(x) and not ops.is_floating_point(x):
x = ops.cast(x, mstype.float32)
return _get_cache_prim(ops.Reciprocal)()(x)

View File

@ -2456,7 +2456,7 @@ def pad(input_x, padding, mode='constant', value=None):
ValueError: If mode is not "constant" and value not None.
Supported Platforms:
``Ascend`` ``GPU`` ``CPU``
``GPU`` ``CPU``
Examples:
>>> import mindspore as ms