tf.nn.conv2d用法简介

tf.nn.conv2d（1.3.0版本）

#-*- coding:utf-8 -*-import numpy as npimport tensorflow as tf # 生成大小为[1,5,5,3]的输入图像batch = 1in_height = in_width = 5in_channels = 3input = tf.constant(1.0, shape=[batch, in_height, in_width, in_channels])# 生成大小为[3,3,3,1]的卷积核filter_width = filter_height = 3filter = tf.constant(1.0, shape=[filter_width, filter_height, in_channels, 1])# 卷积运算output1 = tf.nn.conv2d(input, filter, strides=[1,2,2,1], padding="SAME", name="conv1")output2 = tf.nn.conv2d(input, filter, strides=[1,2,2,1], padding="VALID", name="conv2")# 显示输出图像尺寸print("output1: %s" %np.shape(output1))print("output2: %s" %np.shape(output2))

# 输出结果output1: (1, 3, 3, 1)output2: (1, 2, 2, 1)