Question

python 如何得到矩阵的大小，纵向和横向的维度

fc2 = tf.reshape(fc1, [-1, weights['wd2'].get_shape().as_list()[0]])
afc1=pre_alex_net(x, weights, biases, dropout)
afc1= tf.reshape(afc1,[1024,-1])
fc2 =tf.add(tf.matmul(afc1, fc2),biases['bd2'])

就是现在求fc2的横向和纵向，两个维度怎么写。

Answer 1

(h,w) = fc2.shape()