tensorflow pooling

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/luoganttcc/article/details/82924084

import tensorflow as tf

temp = [0., 0., 1., 0., 0., 0., 1.5, 2.5]

# Reshape the tensor to be 3 dimensions.
values = tf.reshape(temp, [1, 8, 1])


# Use an averaging pool on the tensor.
p_avg = tf.nn.pool(input=values,
    window_shape=[2],
    pooling_type="AVG",
    padding="SAME")

# Use max with this pool.
p_max = tf.nn.pool(input=values,
    window_shape=[2],
    pooling_type="MAX",
    padding="SAME")

session = tf.Session()

# Print our tensors.
print("VALUES")
print(session.run(values))
print("POOL")
print(session.run(p_avg))
print("POOL MAX")
print(session.run(p_max))

session.close()

import tensorflow as tf
a=tf.constant([
        [[1.0,2.0,3.0,4.0],
        [5.0,6.0,7.0,8.0],
        [8.0,7.0,6.0,5.0],
        [4.0,3.0,2.0,1.0]],
        [[4.0,3.0,2.0,1.0],
         [8.0,7.0,6.0,5.0],
         [1.0,2.0,3.0,4.0],
         [5.0,6.0,7.0,8.0]]
    ])


a=tf.reshape(a,[1,4,4,2])

pooling=tf.nn.max_pool(a,[1,2,2,1],[1,1,1,1],padding='SAME')
with tf.Session() as sess:
    print("image:")
    image=sess.run(a)
    print (image)
    print("reslut:")
    result=sess.run(pooling)
    
    print (result)
  

链接1
链接2