本文主要是介绍Tensorflow常用函数-tensorflow.nn.conv2d,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
tf.nn.conv2d(input=None, filters=None, strides=None, padding=None, use_cudnn_on_gpu=True, data_format=None, name=None)
input:需要计算卷积的图像数据 [batch,height,width,channels]
filters:卷积核 [height,width,channels,count]
strides:步长
padding:对卷积结果进行扩充 格式为:[[0,0],[0,0],[0,0],[0,0]]
use_cudnn_on_gpu:是否使用GPU加速
data_format:数据格式,例如"NHWC","NCHW",主要是参数channels位置不同
name:指定该操作的名字
特别的,当padding='SAME',strides=1可以得到输入输出同大小的卷积层,其中padding的具体数量由Tensorflow自动计算并完成操作。
这篇关于Tensorflow常用函数-tensorflow.nn.conv2d的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!