Tensorflow（一）--用tensorflow实现卷积层和池化层的前向传播

版权声明：本文为博主编写文章，未经博主允许转载，转载请注明出处： https://blog.csdn.net/qq_39742013/article/details/83048679

import tensorflow as tf
weights=tf.get_variable("weights",[5,5,3,16],initializer=tf.truncated_normal_initializer(stddev=0.1))
#偏值项
bias=tf.get_variable("biases",[16],initializer=tf.constant_initializer(0.1))
#定义网络
conv=tf.nn.conv2d(input,weights,[1,1,1,1],'SAME')
bias_add=tf.nn.bias_add(conv,bias)
actived_conv=tf.nn.relu(bias_add)

#池化层实现
pool=tf.nn.max_pool(actived_conv,[1,2,2,1],[1,2,2,1],'SAME')