mindspore.mint.imag

View Source On Gitee
mindspore.mint.imag(input) → Tensor[source]

Return a new tensor containing the imaginary values of the input tensor. The returned tensor and input tensor share the same underlying storage.

Note

  • Only support Pynative mode.

  • Only support complex64 and complex128 tensors.

Parameters

input (Tensor) – The input tensor, the data type must be complex64 or complex128.

Returns

Tensor, the shape is same as input. The data type is float32 if input is complex64, float64 when input is complex128.

Raises

  • TypeError – If dtype of input is not complex64 or complex128.

  • ValueError – If input tensor has no storage info.

Supported Platforms:

Ascend

Examples

>>> 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