tf.compat.v1.foldr

foldr по списку тензоров, распакованных из elems по размерности 0.

tf.compat.v1.foldr(
    fn, elems, initializer=None, parallel_iterations=10, back_prop=True,
    swap_memory=False, name=None
)

Этот оператор foldr многократно применяет вызываемую функцию fn к последовательности элементов от последнего к первому. Элементы состоят из тензоров, распакованных из elems. Функция вызова fn принимает два тензора в качестве аргументов. Первый аргумент — это накопленное значение, вычисленное из предыдущего вызова fn, а второй — значение в текущей позиции elems. Если initializer равно None, elems должен содержать по крайней мере один элемент, и его первый элемент используется в качестве инициализатора.

Предположим, что elems распаковывается в values, список тензоров. Форма тензора результата — fn(initializer, values[0]).shape.

Этот метод также позволяет использовать многоарность elems и вывод fn. Если elems представляет собой (возможно, вложенный) список или кортеж тензоров, каждый из этих тензоров должен иметь соответствующую первую (распаковку) размерность. Подпись fn может соответствовать структуре elems. То есть, если elems является (t1, [t2, t3, [t4, t5]]), тогда подходящая подпись для fn — это: fn = lambda (t1, [t2, t3, [t4, t5]]):.

Пример:

elems = [1, 2, 3, 4, 5, 6]
sum = foldr(lambda a, x: a + x, elems)
# sum == 21