如何获得Tensorflow估算器model_fn输入形状？

Question

我想使用tensorflow估算器来构建堆叠自动编码器，并使用input_fn批量输入模型。但我不知道如何让动态形状来设定重量。

代码示例如下：

def model_fn(features, labels, mode):   
    W_init = tf.contrib.layers.variance_scaling_initializer()
    w = tf.get_Variable(shape= ???, initializer=W_init  )
    ...