tf.nn.conv2d(input,filter,strides,padding,use_cudnn_on_gpu=True,data_format='NHWC',dilations=[1,1,1,1],name=None)
参数解释:
input,是4D的,它的数据格式可以是 half,bfloat16,float32,float54
维度顺序由data_format决定,默认是[batch,height,width,channels],也就是“NHWC”格式的。也可以设置为“NCHW”格式的,这时候顺序是[batch,channels,height,width]
filter,和input的数据格式必须一样,也是4D,filter维度是[filter_height,filter_width,in_channels,out_channels]
strides是1D的,int的数据格式,代表了对input的每一维上滑动的窗口距离。
The dimension order is determined by the value of ‘data_format’
strides[0]和strides[3]大多数情况是1,这里不明白,需进一步考证。没见过设置成别的形式的。
dilation batch和channel维必须是1,维度顺序同上,由data_frame决定。dilation实现什么有待进一步考证。请大神指教!