| View source on GitHub |
Map the function fn over the elements elems and return the outputs.
tf.keras.backend.map_fn(
fn, elems, name=None, dtype=None
)
| Arguments | |
|---|---|
fn | Callable that will be called upon each element in elems |
elems | tensor |
name | A string name for the map node in the graph |
dtype | Output data type. |
| Returns | |
|---|---|
Tensor with dtype dtype. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/keras/backend/map_fn