Tensorflow：张量的行数和列数

Question

在tensorflow中，如何得出张量的行数和列数？

如果存在如下的重量张量：

W1 = tf.get_variable（“ W1”，shape = [784，512]，                      initializer = tf.contrib.layers.xavier_initializer（））

我希望知道W1的行数和列数，并自动将其保存在某些参数中。

例如， W1_rows <= W1的行数 W1_columns <= W1的列数

（我知道tf.shape或get_shape（）可以打印形状，但是该方法不能用于保存某些参数中的形状（例如W1_rows，W1_columns）

Answer 1

W1 = tf.get_variable("W1", shape=[784, 512],
                 initializer=tf.contrib.layers.xavier_initializer())
W1_rows, W1_columns = W1.get_shape()
print(W1_rows, W1_columns)

784 512

有效。谢谢你，奥斯瓦尔德。