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]
- `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.
Metody publiczne
statyczny <T rozszerza liczbę, U, V rozszerza liczbę> RaggedGather <T, U> | |
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 , Iterowalny < 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`. |
Powroty
- 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.