#tensorflow如何从main中的函数打印变量?

时间:2018-05-22 14:54:21

标签: python tensorflow

我是tensorflow的新手,我正在搜索如何从python中的main函数中打印一个变量。 我的功能是卷积的定义,这是一个卷曲的"图像"在函数内部定义权重,即:

def conv_fn(p_input):
    # Convolutional Layer #1  -------
    weights1 = tf.get_variable("weights1", [3, 3, 3, 32], 
    initializer=tf.random_normal_initializer())
    conv1   = tf.nn.conv2d(
    input   = p_input.astype(np.float32),
    filter  = weights1,
    strides = [1, 1, 1, 1],
    padding = "VALID")
    conv1_relu = tf.nn.relu(conv1)

    return conv1_relu

主要是我想用图像进行卷积并打印出来。我读到我必须定义一个会话并评估变量或者提供它。我迷失在这一部分......这是我的主要内容:

def main( argv ):

    sess = tf.InteractiveSession()
    sess.run( tf.global_variables_initializer() )

    im = [ 1.0, 1.0, 2.0, 5.0, 1.0, 1.0, 2.0, 5.0, 5.0,1.0, 1.0, 2.0, 5.0, 1.0, 1.0, 2.0, 5.0, 5.0,1.0, 1.0, 2.0, 5.0, 1.0, 1.0, 2.0, 5.0, 5.0 ];
    image = np.reshape( im, [-1, 3, 3, 3] )

    con = conv_fn( image )

    print( "This is image", image )

    print("This is con :", sess.run( con, feed_dict ={p_input: image} ))
if __name__ == "__main__":
    tf.app.run()

请告诉我如何打印:

  • 来自函数conv_fn

  • 的权重
  • 卷积的结果

感谢您的帮助。

0 个答案:

没有答案