Тензор потока: количество строк и столбцов тензора

Question

В тензорном потоке, как я могу получить количество строк и столбцов тензорного?

Если есть тензор веса следующим образом:

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

Я надеюсь узнать количество строк и столбцов W1 и автоматически сохранить их в некоторых параметрах.

например, W1_rows <= количество строк W1 W1_columns <= количество столбцов W1 </p>

(я знаю, что 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

Работает.Спасибо, Освальд.