Gather

publiczne zajęcia końcowe . Zbierzcie się

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>
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.
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.

publiczne statyczne Gather.Options BatchDims (Długie BatchDims)

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ś w `params`, z której zbierane są `indeksy`. Domyślnie pierwszy wymiar. Obsługuje indeksy ujemne.
opcje przenosi opcjonalne wartości atrybutów
Zwroty
  • 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:]`.