Я хочу построить модель, в которой уровень GRU будет пропускать соединения на каждом временном шаге. В основном для каждого временного шага у меня есть значение alpha [t]. который определяет погоду, чтобы переслать скрытое состояние этой ячейки или пересылает значение предыдущей ячейки
alpha_t = 1 подразумевает -> пропустить значение, но если alpha_t = 0 -> остановить значение и получить предыдущеезначение ячейки

alpha [t] = 1 => пропустить значение,
, но если alpha [t] =0 -> остановить значение и получить предыдущее значение ячейки
Может кто-нибудь предложить мне способ достижения этого? Я могу сделать это, создав цикл for для каждого временного шага, но затемдействительно замедлит вычисления. моя модель имеет около 300 временных шагов.