python之在 tensorflow 中初始化矩阵

我有 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)) 

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

请您参考如下方法:

如果您想创建一个 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)