我想使用 tf.get_variable
创建一个变量,它应该用一个 numpy 数组初始化。
据我所知,有两种创建变量的方法,tf.Variable
和tf.get_variable
。我们可以轻松地将初始值传递给 tf.Variable
创建的变量。但是似乎没有直接的方法可以对 tf.get_variable
创建的对象做同样的事情。
我现在一直在做的是首先使用 tf.convert_to_tensor
创建一个带有 numpy 数组的张量,然后将这个张量作为 initializer
传递给 tf .get_variable
。但这对我来说似乎毫无效率,因为我有一个非常大的数组作为初始值,然后我必须在我的图中存储这个大数组两次。
因此,我想知道是否有一些更有效的方法来做到这一点?使用 Variable
是避免我上面提到的效率问题的唯一方法吗?
最佳答案
答案是使用 TensorFlow 的函数 tf.constant_initializer(value)
(参见 doc)。
虽然文档说只使用标量值,但您可以传递任何维度的 numpy 数组。
关于python - 如何有效地将初始值传递给 get_variable,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37272603/