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

前端之家收集整理的这篇文章主要介绍了如何在TensorFlow中将Tensor变量初始化为单位矩阵前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

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

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

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

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

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

猜你在找的Python相关文章