您的位置:首页 > 其它

小记tensorflow-1:tf.nn.conv2d 函数介绍

2017-10-13 21:09 483 查看
tf.nn.conv2d函数介绍



Input: 输入的input必须为一个4d tensor,而且每个input的格式必须为float32 或者float64. Input=[batchsize,image_w,image_h,in_channels],也就是[每一次训练的batch数,图片的长,图片的宽,图片的通道数]。

Filter: 和input类似。样式为:[ 卷积核的长,卷积核的宽,输入的通道,输出的通道]

Strides:长度为4的list,且元素类型为int..格式为:[1,x,y,1].表示的是每一维度滑动的的步长。

Padding: 有"Same","VALID" 两种选择。SAME:为要对图像进行补零操作,VALID不要对图像进行补零操作。

use_cudnn_on_gpu: 布尔类型,True, False 两种选择。表示是否用GPU训练。

Name:

表示该操作的名字
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: