tf.raw_ops.RaggedGather

Собирает нерегулярные фрагменты по оси params 0 в соответствии с indices.

Просмотреть псевдонимы

Псевдонимы совместимости для миграции

Дополнительные сведения см. в руководстве по миграции.

tf.compat.v1.raw_ops.RaggedGather

tf.raw_ops.RaggedGather(
    params_nested_splits,
    params_dense_values,
    indices,
    OUTPUT_RAGGED_RANK,
    name=None
)

Формирует выходной RaggedTensor, состоящий из output_dense_values и output_nested_splits, так что:

output.shape = indices.shape + params.shape[1:]
output.ragged_rank = indices.shape.ndims + params.ragged_rank
output[i...j, d0...dn] = params[indices[i...j], d0...dn]

где

params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits) содержит значения, которые необходимо собрать.
indices — плотный тензор с типом данных int32 или int64, указывающий, какие значения необходимо собрать.
output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits) — выходной тензор.

(Примечание: этот оператор C++ используется для реализации оператора Python более высокого уровня tf.ragged.gather, который также поддерживает нерегулярные индексы.)

Аргументы
`params_nested_splits`	Список как минимум из 1 объекта `Tensor` одного типа из: `int32`, `int64`. Тензоры `nested_row_splits`, задающие разбиение на строки для входных данных RaggedTensor `params`.
`params_dense_values`	`Tensor`. `flat_values` для RaggedTensor `params`. На уровне Python термин dense_values был заменён на flat_values, поэтому dense_values — устаревшее имя.
`indices`	`Tensor`. Должен иметь один из следующих типов: `int32`, `int64`. Индексы во внешнем измерении `params` значений, которые необходимо собрать.
`OUTPUT_RAGGED_RANK`	`int` типа `>= 0`. Ранг нерегулярности выходного RaggedTensor. `output_nested_splits` будет содержать указанное количество тензоров `row_splits`. Это значение должно равняться `indices.shape.ndims + params.ragged_rank - 1`.
`name`	Имя операции (необязательно).

Возвращает
Кортеж объектов `Tensor` (output_nested_splits, output_dense_values).
`output_nested_splits`	Список объектов `OUTPUT_RAGGED_RANK` `Tensor` того же типа, что и `params_nested_splits`.
`output_dense_values`	`Tensor`. Имеет тот же тип, что и `params_dense_values`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/raw_ops/RaggedGather