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.