python - 在 tensorflow 中初始化矩阵

Question

我有 6 个矩阵，我的模型将学习它们，我将它们定义如下:

self.R= tf.get_variable('R_',dtype=tf.float32, shape=[6,300 ,300],
           initializer=tf.random_uniform_initializer(maxval=0.1, minval=-0.1))

我需要做的是改变初始化。我想将它们中的每一个初始化为单位矩阵。有人可以帮我吗？

Answer 1

如果您想创建一个 6x300x300 矩阵，其中每个 300x300 数组都是一个单位矩阵，您可以简单地:

import numpy as np;

dimension = 300 
singleIdentityMatrix = np.identity(dimension, dtype= np.float32) 
stackedMatrix = np.dstack( [singleIdentityMatrix] * 6)

并用

传递这个矩阵

self.R = tf.Variable(initial_value = stackedMatrix)