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

[Anonymous]

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

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

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

Вывод операции 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