欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

机器学习杂记,主要记录学习过程中的一些疑问和理解

程序员文章站 2024-03-25 09:48:52
...

最近看tensorflow,试着写 cnn 代码,其中的一个接口 :

def conv2d(input, filter, strides, padding, use_cudnn_on_gpu=True, data_format="NHWC", dilations=[1, 1, 1, 1], name=None)

其中 filter 这个 tensor  [filter_height, filter_width, in_channels, out_channels] 对应的意思就是[卷积核的高度,卷积核的宽度,图像通道数,卷积核个数] 刚开始一直纠结的就是第四维哈