tf.data.experimental.scan  |  TensorFlow v2.16.1 (original) (raw)

tf.data.experimental.scan

A transformation that scans a function across an input dataset. (deprecated)

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.data.experimental.scan

tf.data.experimental.scan(
    initial_state, scan_func
)

This transformation is a stateful relative of tf.data.Dataset.map. In addition to mapping scan_func across the elements of the input dataset,scan() accumulates one or more state tensors, whose initial values areinitial_state.

Args
initial_state A nested structure of tensors, representing the initial state of the accumulator.
scan_func A function that maps (old_state, input_element) to(new_state, output_element). It must take two arguments and return a pair of nested structures of tensors. The new_state must match the structure of initial_state.
Returns
A Dataset transformation function, which can be passed totf.data.Dataset.apply.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.