Как выполнить 2D-заполнение вперед по 0-му измерению в Tensorflow ⇐ Python

[Anonymous]

В качестве примера возьмем следующий тензор:

Код: Выделить всё

tf.constant([
[0,      np.nan, 2, 1],
[np.nan, 3,      3, 4],
])

Я хотел бы реализовать операцию прямого заполнения, например pd.DataFrame.ffill (с осью = 0). Таким образом, результат операции ffill должен быть следующим:

Код: Выделить всё

[
[0, np.nan, 2, 1]
[0, 3,      3, 4]
]

Однако мне не ясно, как мне добиться такого тензорного преобразования. Я нашел следующий вопрос, касающийся этого случая 1D:
https://stackoverflow.com/a/52818310/1613983
Однако , переменнаяvalues в этой реализации представляет собой разреженное представление входных данных, которое отлично работает в одном измерении, но мне не удалось преобразовать это в 2D-случай.

Подробнее здесь: https://stackoverflow.com/questions/792 ... tensorflow