def get_en_z_variables(self):
weights = {
'e1': tf.Variable(tf.random_normal([4, 4, 1, 64], stddev=0.02), name='enz_w1'),
'e2': tf.Variable(tf.random_normal([4, 4, 64, 128], stddev=0.02), name='enz_w2'),
##z
'e3': tf.Variable(tf.random_normal([128 * 7 * 7, 64], stddev=0.02), name='enz_w3')
}
biases = {
'eb1': tf.Variable(tf.zeros([64]), name='enz_b1'),
'eb2': tf.Variable(tf.zeros([128]), name='enz_b2'),
##z
'eb3': tf.Variable(tf.zeros([64]), name='enz_b3')
}
return weights, biases
评论列表
文章目录