我有 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)
关于python - 在 tensorflow 中初始化矩阵,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46169740/