python - 如何有效地将初始值传递给 get_variable

Question

我想使用 tf.get_variable 创建一个变量，它应该用一个 numpy 数组初始化。

据我所知，有两种创建变量的方法，tf.Variable 和tf.get_variable。我们可以轻松地将初始值传递给 tf.Variable 创建的变量。但是似乎没有直接的方法可以对 tf.get_variable 创建的对象做同样的事情。

我现在一直在做的是首先使用 tf.convert_to_tensor 创建一个带有 numpy 数组的张量，然后将这个张量作为 initializer 传递给 tf .get_variable。但这对我来说似乎毫无效率，因为我有一个非常大的数组作为初始值，然后我必须在我的图中存储这个大数组两次。

因此，我想知道是否有一些更有效的方法来做到这一点？使用 Variable 是避免我上面提到的效率问题的唯一方法吗？

Answer 1

答案是使用 TensorFlow 的函数 tf.constant_initializer(value)(参见 doc)。

虽然文档说只使用标量值，但您可以传递任何维度的 numpy 数组。