как перебирать список тензорного потока, ссылаясь как на индекс, так и на отдельный тензор - PullRequest
0 голосов
/ 01 ноября 2018

Я попробовал следующий код, использовал 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)]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...