tf.ragged.map_flat_values
| Просмотреть исходный код на GitHub |
Применяет op к значениям одного или нескольких RaggedTensors.
tf.ragged.map_flat_values(
op, *args, **kwargs
)
Заменяет все RaggedTensor в args или kwargs на соответствующий flat_values тензор, а затем вызывает op. Возвращает RaggedTensor, построенный из входных RaggedTensor и значения, возвращаемого op.
Если входные аргументы содержат несколько RaggedTensors, то они должны иметь одинаковые nested_row_splits.
Примеры:
rt = ragged.constant([[1, 2, 3], [], [4, 5], [6]]) ragged.map_flat_values(tf.ones_like, rt).eval().tolist() [[1, 1, 1], [], [1, 1], [1]] ragged.map_flat_values(tf.multiply, rt, rt).eval().tolist() [[1, 4, 9], [], [16, 25], [36]] ragged.map_flat_values(tf.add, rt, 5).eval().tolist() [[6, 7, 8], [], [9, 10], [11]]
| Аргументы | |
|---|---|
op | Операция, которая должна быть применена к RaggedTensor flat_values. op обычно является поэлементной операцией (например, math_ops.add), но может быть использована любая операция, сохраняющая размер внешнего измерения. То есть, размерность значения, возвращаемого op , должна совпадать с размерностью shape[0] тензоров RaggedTensor . |
*args | Аргументы для op . |
**kwargs | Именованные аргументы для op . |
| Возвращаемое значение | |
|---|---|
RaggedTensor , размер внешнего измерения которого соответствует размерности всех входных RaggedTensor . |
| Исключения | |
|---|---|
ValueError | Если args не содержит RaggedTensors, или если размер внешнего измерения входных RaggedTensor не совпадает. |
© 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/r1.15/api_docs/python/tf/ragged/map_flat_values