如何在TensorFlow中将Tensor变量初始化为单位矩阵

Question

如何将Tensor T初始化为单位矩阵？

以下将T初始化为784×784的零矩阵.

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

但我找不到行为符合要求的tf.fn.如何才能做到这一点？

Answer 1

您要查找的tf.fn称为tf.eye。因此，最简洁的答案是

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

注意：将其放在tf.Variable中可以将可识别的权重初始化为身份，但可以对其进行更改。如果您实际上只是想要一个恒等矩阵的常数，则只需使用

T = tf.eye(size)

Answer 2

您实际上可以将numpy数组作为initial_value的参数传递,因此tf.Variable(initial_value = np.identity(784))应该执行您想要执行的操作.