Spec-Zone.ru › TensorFlow

tf.compat.v1.foldl

foldl для списка тензоров, извлечённых из elems по измерению 0.

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

Этот оператор foldl многократно применяет вызываемый объект fn к последовательности элементов от первого до последнего. Элементы состоят из тензоров, извлечённых из elems по измерению 0. Вызываемый объект 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 = tf.constant([1, 2, 3, 4, 5, 6])
sum = foldl(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/foldl

Spec-Zone.ru

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