complex_util.py 文件源码

def get_variable_c( name, shape, initializer=None ):
    re = tf.get_variable(name+'_re', shape=shape, initializer=initializer)
    im = tf.get_variable(name+'_im', shape=shape, initializer=initializer)
    return tf.complex(re,im, name=name)


#get unit complex numbers in polar form