机器学习杂记,主要记录学习过程中的一些疑问和理解
程序员文章站
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] 对应的意思就是[卷积核的高度,卷积核的宽度,图像通道数,卷积核个数] 刚开始一直纠结的就是第四维哈