Spec-Zone.ru › TensorFlow

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]]):.

Аргументы
fn Вызываемый объект, который необходимо выполнить.
elems Тензор или последовательность тензоров (возможно, вложенная), каждый из которых будет распакован по первому измерению. Вложенная последовательность полученных срезов станет первым аргументом для fn.
initializer (необязательно) Тензор или последовательность тензоров (возможно, вложенная), используемые в качестве начального значения аккумулятора.
parallel_iterations (необязательно) Количество итераций, которые можно выполнять параллельно.
back_prop (необязательно) Значение True включает поддержку обратного распространения.
swap_memory (необязательно) Значение True включает обмен памятью между GPU и CPU.
name (необязательно) Префикс имени для возвращаемых тензоров.
Возвращает
Тензор или последовательность тензоров (возможно, вложенная), полученная в результате последовательного применения fn к списку тензоров, распакованных из elems, от последнего к первому.
Вызывает исключения
TypeError если fn не является вызываемым объектом.
Пример
elems = [1, 2, 3, 4, 5, 6]
sum = foldr(lambda a, x: a + x, elems)
# sum == 21

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/foldr

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API