张量-复数操作函数

tf.complex(real,imag,name = None),该函数用于将两实数转化为复数。

示例代码如下:

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

real = tf.constant([2.25,3.25],dtype = tf.float64)
imag = tf.constant([4.75,5.75],dtype = tf.float64)

with tf.Session() as sess:
    print(sess.run(tf.complex(real,imag)))

tf.math.abs(x,name = None),用于计算复数的绝对值,即长度。

复数的绝对值,也就是复数的模,它的计算方式为将复数的实部与虚部的平方和,取平方根

示例代码如下:

import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

a = tf.constant([[-2.25+4.75j],[-3.25+5.75j]])

with tf.Session() as sess:
    print(sess.run(tf.math.abs(a)))

tf.conj(input,name = None)用于计算共轭复数

tf.imag(input,name = None)用于取得复数的虚部

tf.real(input,name = None)用于取得复数的实部

tf.flt(input,name = None)用于计算一维的离散傅里叶变换,输入类型为complex64