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