Python – tensorflow.math.imag()

概述

tensorflow.math.imag() 是 tensorflow 的一个函数，用于获取输入复数所表示的虚部。

语法

tensorflow.math.imag(x, name=None)

参数

• x：必选参数，指定输入的复数张量。
• name：可选参数，指定操作的名称。

返回值

返回一个表示 x 中每个元素虚数部分的张量。

示例

下面是使用 tensorflow.math.imag() 函数的示例代码：

import tensorflow as tf
a = tf.constant([1+2j, 3+4j])
b = tf.constant([[5+6j, 7+8j], [9+10j, 11+12j]])
c = tf.math.imag(a)
d = tf.math.imag(b)
print(c)
print(d)

输出：

tf.Tensor([2. 4.], shape=(2,), dtype=float32)
tf.Tensor(
[[ 6.  8.]
 [10. 12.]], shape=(2, 2), dtype=float32)

在这个示例中，我们使用 tensorflow.math.imag() 函数来获取输入张量中每个复数的虚部。第一个示例中，a 是一个一维复数张量，c 是一个包含 a 中每个元素虚部的一维张量。在第二个示例中，b 是一个二维复数张量，d 是一个包含 b 中每个元素虚部的二维张量。

注意

• tensorflow.math.imag() 函数只能处理复数张量，如果输入的张量不是复数张量，则会抛出 InvalidArgumentError 异常。
• 实数的虚部为 0，因此如果输入为实数，则函数将返回一个全为 0 的张量。