тензорная операция в тф2

Question

Я хотел бы получить вывод, применяя функцию к тензору в модели. Я пробовал оба тензорных оператора и numpy функции, но оба не удалось. Я применил бы операцию:

def output2_new_from_output1_result(output1):
  start_new = np.zeros((1, output1.shape[1]))

  for step in range(output1.shape[1]):
      cans = np.where(output1[:, step] != 0)[0]
      if len(cans) > 0:
          for i in cans:
              if output1[i, step] == 1 and output1[i, step-1] == 0:
                  start_new[:, step] = 1

output1 - это выход в модели, и я хотел бы получить другой выход из этого output1.

Записная книжка Colab по этому поводу link