mindspore.ops.Imag
- class mindspore.ops.Imag[源代码]
返回包含输入Tensor的虚部。如果输入为实数,则返回零。
- 输入:
input (Tensor) - 要计算的输入Tensor。
- 输出:
Tensor,shape与 input 相同。
- 异常:
TypeError - 如果 input 不是Tensor。
- 支持平台:
GPU
CPU
样例:
>>> x = Tensor(np.asarray(np.complex(1.3+0.4j)), mindspore.complex64) >>> imag = ops.Imag() >>> output = imag(x) >>> print(output) 0.4