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

发布时间：2020-12-16 22:40:05 所属栏目：Python 来源：网络整理

导读：如何将Tensor T初始化为单位矩阵？以下将T初始化为784×784的零矩阵. T = tf.Variable(tf.zeros([784,784])) 但我找不到行为符合要求的tf.fn.如何才能做到这一点？最佳答案实际上,您可以将numpy数组作为initial_value的参数传递,因此tf.Variable(initial_

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

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

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

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

最佳答案

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!