RaggedGather

publiczna klasa końcowa RaggedGather

Zbierz poszarpane plasterki z osi „0” „params” zgodnie z „indeksami”.

Wysyła wyjście `RaggedTensor` złożone z `output_dense_values` i `output_nested_splits`, takie jak: 

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]
gdzie
  • `params = ragged.from_nested_row_splits(params_dense_values, params_nested_splits)` udostępnia wartości, które należy zebrać.
  • `indeksy` to gęsty tensor o typie `int32` lub `int64`, wskazujący, które wartości mają być zbierane.
  • `output = ragged.from_nested_row_splits(output_dense_values, Output_nested_splits)` jest tensorem wyjściowym.
(Uwaga: ta operacja w języku C++ służy do implementacji operacji `tf.ragged.gather` w języku Python wyższego poziomu, która obsługuje również niewyrównane indeksy.)

Metody publiczne

statyczny <T rozszerza liczbę, U, V rozszerza liczbę> RaggedGather <T, U>
utwórz ( zakres zakresu, Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> indeksy, Long OUTPUTRAGGEDRANK)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację RaggedGather.
Wyjście <U>
wyjścioweDenseValues ​​()
„Wartości_płaskie” dla zwróconego RaggedTensor.
Lista< Wyjście <T>>
wyjścieNestedSplits ()
Tensory „nested_row_splits”, które definiują podział wierszy dla zwróconego RaggedTensor.

Metody dziedziczone

Metody publiczne

public static RaggedGather <T, U> create ( zakres zakresu , Iterable< Operand <T>> paramsNestedSplits, Operand <U> paramsDenseValues, Operand <V> indeksy, Long OUTPUTRAGGEDRANK)

Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację RaggedGather.

Parametry
zakres aktualny zakres
parametryNestedSplits Tensory „nested_row_splits” definiujące podział wierszy dla danych wejściowych „params” RaggedTensor.
paramsDenseValues „Wartości_płaskie” dla „parametrów” RaggedTensor. Nastąpiła zmiana terminologii na poziomie Pythona z wartości gęste na wartości płaskie, więc wartości gęste są przestarzałą nazwą.
indeksy Wskaźniki w najbardziej zewnętrznym wymiarze „parametrów” wartości, które należy zebrać.
WYJŚCIOWY RANK Nierówna ranga wyniku RaggedTensor. `output_nested_splits` będzie zawierać tę liczbę tensorów `row_splits`. Ta wartość powinna być równa `indices.shape.ndims + params.ragged_rank - 1`.
Zwroty
  • nowa instancja RaggedGather

publiczne dane wyjściowe <U> dane wyjścioweDenseValues ​​()

„Wartości_płaskie” dla zwróconego RaggedTensor.

public List< Output <T>> OutputNestedSplits ()

Tensory „nested_row_splits”, które definiują podział wierszy dla zwróconego RaggedTensor.