Как инициализировать переменную тензора как идентификационную матрицу в TensorFlow

Question

Как инициализировать тензор T как единую матрицу?

Следующее инициализирует T как матрицу нулей 784 на 784.

T = tf.Variable(tf.zeros([784, 784])) 

Но я не могу найти tf.fn, который ведет себя по мере необходимости. Как это может быть сделано?

Answer 1

Фактически вы можете передавать массивы numpy в качестве аргумента для initial_value, поэтому tf.Variable(initial_value = np.identity(784)) должен делать то, что вы намереваетесь сделать.

Answer 2

Не хотите устанавливать numpy только для np.identity?
Вот tensorflow-единственный вариант:

T = tf.Variable(tf.diag(tf.ones(size))) 

Answer 3

tf.fn вы ищете называется tf.eye. Таким образом, наиболее емким ответ

T = tf.Variable(tf.eye(size)) 

Примечание: помещая это в tf.Variable инициализирует изучаемым веса к единице, но допускает, что оно может быть изменено. Если вы на самом деле просто хотите постоянную единичную матрицу, а затем просто использовать

T = tf.eye(size)