如何将密集层从Tensorflow 1迁移到Tensorflow 2?

时间:2020-07-05 21:28:49

标签: python tensorflow keras

如何将这一层迁移到tf2

\\2

我发现占位符现在是“输入”,并且我将tense层用于了tf2

我尝试过:

.sub-menu { 
    width: 480px;
    display: grid; 
    grid-template-columns: 1 1; 
}


.sub-menu-columns ul.sub-menu li {
float: left;
width: 50%;
}



.sub-menu-columns ul.sub-menu li:nth-child(odd) {
float: left;
}


.sub-menu-columns ul.sub-menu li:nth-child(even) {
float: right;
}

如果使用此错误,我会得到

observations = tf.placeholder(tf.float32,[None, OBSERVATIONS_SIZE])

h = tf.layers.dense(
     observations,
     units=hidden_layer_size,
     activation=tf.nn.relu,
     kernel_initializer=tf.contrib.layers.xavier_initializer()
)

在这种情况下,我应该如何使用占位符/输入?

1 个答案:

答案 0 :(得分:1)

Keras层不用作tf.layers,它们是可调用的,而不是将张量作为第一个参数传递,因此它应该是:

observations = tf.keras.Input(
    shape = [ None, OBSERVATIONS_SIZE ],
    dtype = tf.float32
    )

h = tf.keras.layers.Dense(
     units=hidden_layer_size,
     activation='relu',
     kernel_initializer = 'glorot_uniform'
     )(observations)