我正在尝试创建一个常数张量并从图中的多个位置访问它。
目前,我正在任何函数(文件末尾)之外定义常量张量
##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).
是否有一种方法可以定义常数张量,以便可以从图中的任何位置访问它们?