tensorflow - 如何迭代引用索引和单个张量的tensorflow tensor_list

Question

我尝试了以下代码，使用了 tf.map_fn，但仍然出现错误：“张量对象仅在启用急切执行时才可迭代。要迭代此张量，请使用 tf.map_fn。”

def get_tensor_A(i):
    return mask[i]   #another tensor returned

def get_tensor_B(x):
    return tf.nn.softmax(x, 1)

def final_rt(i, inp):
    return get_tensor_A(idx)*get_tensor_B(inp)

tensor_2_list = [ tf.map_fn(lambda x: final_rt(x(0), x(1)), (idx, inp))  for idx, inp in enumerate(tensor_1_list)]

标签： tensorflowlambdaenumerate