从张量流中的多个函数访问常量

时间:2019-04-15 09:22:32

标签: python tensorflow

我正在尝试创建一个常数张量并从图中的多个位置访问它。
目前,我正在任何函数(文件末尾)之外定义常量张量

##GLOBAL BASE FUNCTIONS - DEFINED OUTSIDE OF ANY FUNCTION###
BASE_FUNCTIONS = []

for base in range(NUMBER_OF_BASE_FUNCTIONS,0,-1):
    BASE_FUNCTIONS.append(tf.constant(SOME_NP_ARRAY)

然后通过多种功能使用它,例如:

    def some_function():
        base_functions_output = fc_layer * BASE_FUNCTIONS

但是我遇到了以下错误

ValueError: Tensor("tower_0/transform_net1/tconv1/ExpandDims:0", shape=(8192, 10, 1), dtype=float32, device=/device:GPU:0) must be from the same graph as Tensor("tower_0/transform_net1/tconv1/mul/y:0", shape=(10, 40), dtype=float32).

是否有一种方法可以定义常数张量,以便可以从图中的任何位置访问它们?

0 个答案:

没有答案