mindspore.mint.imag

mindspore.mint.imag(input) → Tensor

返回一个新tensor，包含输入tensor的虚部。

返回的tensor和输入tensor共享相同的底层存储。

说明

• 仅支持Pynative模式。

• 仅支持complex64和complex128类型的tensor。

参数：

• input (Tensor) - 输入tensor。数据类型只支持complex64和complex128。

返回：

Tensor，输出shape与 input 相同。如果输入为complex64，则输出为float32，如果输入为complex128，则输出为float64。

异常：

• TypeError - 如果 input 的数据类型不是complex64或complex128。

• ValueError - 如果输入tensor没有存储信息。

支持平台：

Ascend

样例：

>>> import mindspore
>>> from mindspore import Tensor, mint, ops, context
>>> context.set_context(mode=context.PYNATIVE_MODE, device_target="Ascend")
>>> real = Tensor([1.1, 2.1, 3.1], mindspore.float32)
>>> imag = Tensor([4.1, 5.1, 6.1], mindspore.float32)
>>> x = ops.Complex()(real, imag)
>>> output = mint.imag(x)
>>> print(output)
[4.1 5.1 6.1]
>>> print(output.dtype)
Float32
>>> real = Tensor([1.1, 2.1, 3.1], mindspore.float64)
>>> imag = Tensor([4.1, 5.1, 6.1], mindspore.float64)
>>> x = ops.Complex()(real, imag)
>>> output = mint.imag(x)
>>> print(output)
[4.1 5.1 6.1]
>>> print(output.dtype)
Float64