jax.numpy.imag

jax.numpy.imag(val, /)

返回复数参数的虚部。

LAX 后端实现 numpy.imag()。

原始文档字符串如下。

参数：

val (array_like) – 输入数组。

返回值：

out – 复数参数的虚数部分。如果 `val` 为实数，则使用 `val` 的类型作为输出类型。如果 `val` 包含复数元素，则返回类型为浮点数。

返回类型：

ndarray 或标量