tensorflow 输出numpy显示维度

代码

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)