将tensorflow 1转换为tensorflow 2

Question

我正在将代码从tf v1转换为tf v2。代码的背景是针对自动编码器的，之后我需要使用输出的图层。这是示例代码。将NN层及其相关输出替换为tensorflow v2的最佳方法是什么？

def sample_gmm(samples_s, z_dim, reuse):
    
    mean_pz = tf.layers.dense(inputs=samples_s, units=z_dim, activation=None,
                         kernel_initializer=tf.random_normal_initializer(stddev=0.05), name= 'layer_1_' + 'mean_dec_z', reuse=reuse)
    log_var_pz = tf.zeros([tf.shape(samples_s)[0],z_dim])
    
    # Avoid numerical problems
    log_var_pz = tf.clip_by_value(log_var_pz,-15.0,15.0)
    
    return mean_pz, log_var_pz