代码
import tensorflow as tf
import numpy as np
class Net:
def __init__(self):
self.x = tf.placeholder(dtype=tf.float32,shape=[None,None,None,3])
self.forward()
def forward(self):
self.conv1 = tf.nn.leaky_relu(tf.layers.conv2d(self.x, 10, kernel_size=[3,3], strides=[1,1], padding="valid"))
if __name__ == '__main__':
net = Net()
init = tf.global_variables_initializer()
with tf.Session() as sess:
sess.run(init)
x = np.random.randn(1,12,12,3)
print(x.shape)
conv1 = sess.run(net.conv1,feed_dict={net.x: x})
print(np.array(conv1).shape)
输出
(1, 12, 12, 3)
(1, 10, 10, 10)