我是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
卷积的结果
感谢您的帮助。