Zbierz wycinki z osi `params`, `osi` zgodnie z `indeksami`.
„indeksy” muszą być tensorem liczby całkowitej dowolnego wymiaru (zwykle 0-D lub 1-D). Tworzy tensor wyjściowy o kształcie `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[oś + 1:]` gdzie:
# Scalar indices (output is rank(params) - 1).
output[a_0, ..., a_n, b_0, ..., b_n] =
params[a_0, ..., a_n, indices, b_0, ..., b_n]
# Vector indices (output is rank(params)).
output[a_0, ..., a_n, i, b_0, ..., b_n] =
params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
# Higher rank indices (output is rank(params) + rank(indices) - 1).
output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]

Należy pamiętać, że na procesorze, jeśli zostanie znaleziony indeks spoza limitu, zwracany jest błąd. Na GPU, jeśli zostanie znaleziony indeks poza granicami, w odpowiedniej wartości wyjściowej zostanie zapisane 0.
Zobacz także `tf.batch_gather` i tf.gather_nd
.
Klasy zagnieżdżone
klasa | Zbierz.Opcje | Opcjonalne atrybuty dla Gather |
Metody publiczne
Wyjście <T> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
statyczne Zbierz.Opcje | BatchDims (Długie BaterieDims) |
statyczny <T, U rozszerza liczbę, V rozszerza liczbę> Zbierz <T> | |
Wyjście <T> | wyjście () Wartości z `params` zebrane z indeksów podanych przez `indices`, o kształcie `params.shape[:axis] + indices.shape + params.shape[oś + 1:]`. |
Metody dziedziczone
Metody publiczne
publiczne wyjście <T> asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
public static Zbierz <T> utwórz (zakres zakresu , parametry argumentu <T>, indeksy argumentu <U>, oś argumentu <V>, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację Gather.
Parametry
zakres | aktualny zakres |
---|---|
parametry | Tensor, z którego mają być zbierane wartości. Musi mieć co najmniej rangę „oś + 1”. |
indeksy | Tensor indeksu. Musi należeć do zakresu „[0, params.shape[oś])”. |
oś | Oś w `params`, z której zbierane są `indeksy`. Domyślnie pierwszy wymiar. Obsługuje indeksy ujemne. |
opcje | przenosi opcjonalne wartości atrybutów |
Powroty
- nowa instancja Gather
publiczne wyjście <T> wyjście ()
Wartości z `params` zebrane z indeksów podanych przez `indices`, o kształcie `params.shape[:axis] + indices.shape + params.shape[oś + 1:]`.