Tensorflow学习笔记:卷积层的实现

tensorflow中主要是用到了tf.nn.conv2d()这个函数,来进行卷积的计算,函数常用的参数如下:
tf.nn.conv2d(input,filter,strides,padding)
input : 一个四维的输入,第一维是batch维,后三维是输入矩阵的大小。
filte : 一个四维的过滤器,前两维是过滤器尺寸一般设置为3*3或者5*5,第三维要与输入矩阵的深度保持一致,第四维是过滤器深度,表示了经过卷积计算后单位矩阵的深度。
strides : 四个元素的列表,分别表示在batch,输入矩阵的长宽,输出单位矩阵的深度上过滤器移动的步长。第一个与最后一个元素一般都是1
padding : 使用两个字符串参数,“SAME”表示全0填充,”VALID”表示不进行全0填充

接下来给出一个使用的实例

import tensorflow as tf

input=tf.truncated_normal(shape=[1,5,5,3],stddev=0.1,name="input")
filter=tf.Variable(tf.truncated_normal(shape=[3,3,3,16],stddev=0.1),name="filter")

conv=tf.nn.conv2d(input,filter,strides=[1,1,1,1],padding="VALID")
result=tf.nn.bias_add(conv,tf.random_normal(shape=[16],stddev=0.1))
result=tf.nn.tanh(result)
#使用tf.nn.bias_add加上偏置项后,激活函数去线性化

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(result))

参考:Tensorflow 实战Google深度学习框架

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值