TensorFlow.js tf.real() 函数

tf.real() 函数是 TensorFlow.js 提供的一个用于计算复数的实部的函数。它接受一个类型为 tf.Tensor<tf.complex64> 的张量作为参数，并返回一个类型为 tf.Tensor<tf.float32> 的张量，该张量包含了输入张量的实部。

语法

tf.real<T extends tf.Tensor<tf.complex64>>(input: T): tf.Tensor<tf.float32>

参数

• input：一个类型为 tf.Tensor<tf.complex64> 的输入张量。

返回值

• 一个类型为 tf.Tensor<tf.float32> 的张量，该张量的形状与输入张量的形状相同，它包含了输入张量的实部。

示例

const input = tf.complex([1, 2, 3, 4], [-5, 6, -7, 8]);  // 创建一个复数张量
const realPart = tf.real(input);  // 计算该张量的实部

realPart.print();  // 打印实部张量

// 输出：
// Tensor
//  [ 1, 2, 3, 4 ],
//  dtype: 'float32'

在上面的示例中，我们创建了一个复数张量 input，它包含了 4 个复数。然后，我们使用 tf.real() 函数计算了该张量的实部，并将结果存储在名为 realPart 的变量中。我们采用 print() 方法打印了结果，结果展示了输入张量的实部。

注意事项

• 输入张量必须是类型为 tf.Tensor<tf.complex64> 的复数张量。
• 输出张量的类型是 tf.Tensor<tf.float32>，表示实数。
• 输出张量的形状与输入张量的形状相同。
• 当输入张量中包含纯虚数时，其实部为 0。

结论

tf.real() 函数是一个非常有用的函数，它可以方便地计算任意复数张量的实部。当需要从复数张量中提取实部时，该函数可以是一个非常有用的工具。