RaggedGather

публичный финальный класс RaggedGather

Соберите неровные фрагменты из оси `0` params` в соответствии с `index`.

Выводит выходные данные 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)` предоставляет значения, которые необходимо собрать.
  • Индексы — это плотный тензор с типом dtype int32 или int64, указывающий, какие значения следует собирать.
  • `output = ragged.from_nested_row_splits(output_dense_values, output_nested_splits)` — выходной тензор.
(Примечание. Эта операция C++ используется для реализации операции Python более высокого уровня `tf.ragged.gather`, которая также поддерживает неровные индексы.)

Публичные методы

static <T расширяет число, U, V расширяет число> RaggedGather <T, U>
create (Область видимости , Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> индексы, Long OUTPUTRAGGEDRANK)
Фабричный метод для создания класса, обертывающего новую операцию RaggedGather.
Выход <U>
выходные значенияDenseValues ​​()
Flat_values ​​для возвращенного RaggedTensor.
Список< Выход <T>>
выводNestedSplits ()
Тензорыnested_row_splits, которые определяют разделение строк для возвращаемого RaggedTensor.

Унаследованные методы

Публичные методы

public static RaggedGather <T, U> create (область действия , Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, индексы Operand <V>, Long OUTPUTRAGGEDRANK)

Фабричный метод для создания класса, обертывающего новую операцию RaggedGather.

Параметры
объем текущий объем
параметрыNestedSplits Тензоры `nested_row_splits`, которые определяют разделение строк для входных данных `params` RaggedTensor.
параметрыDenseValues Flat_values ​​для RaggedTensor params. На уровне Python произошло изменение терминологии с Densent_values ​​на Flat_values, поэтому Densent_values ​​— устаревшее имя.
индексы Индексы во внешнем измерении «params» значений, которые необходимо собрать.
ВЫХОДНОЙ ТРАГГЕДРАНК Неровный ранг выходного RaggedTensor. `output_nested_splits` будет содержать это количество тензоров `row_splits`. Это значение должно равняться `indices.shape.ndims + params.ragged_rank - 1`.
Возврат
  • новый экземпляр RaggedGather

публичный вывод <U> outputDenseValues ​​()

Flat_values ​​для возвращенного RaggedTensor.

общедоступный список< Вывод <T>> выходNestedSplits ()

Тензорыnested_row_splits, которые определяют разделение строк для возвращаемого RaggedTensor.