tensorflow conv2d

2020-03-12  本文已影响0人  空谷幽兰_c411

tf.nn.conv2d:
包含四个常用的参数:
input:需要做卷积的图像,需要是一个Tensor,具有[batch_size,in_height,in_width,in_channel]这样的shape
filter:卷积核,要求是一个Tensor,具有[filter_height,filter_width,in_channel,out_channel]
stride:卷积时在图像每一维的步长
padding:string类型,只能是SAME或valid

上一篇 下一篇

猜你喜欢

热点阅读