XlaConcatND

publiczne zajęcia finałowe XlaConcatND

Łączy tensor wejściowy we wszystkich wymiarach.

Operacja scalająca dzieli tensor wejściowy na podstawie danego atrybutu num_splits, opcjonalnie usuwa dopełnienia i zwraca scalony tensor bez dopełnień.

Ta operacja może zostać wygenerowana poprzez most TPU.

Na przykład, z tensorem `input`:

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
 
`num_splits`:
[2, 2]
 
i `paddings`:
[1, 1]
 
oczekiwane `outputs` to:
[[0, 1, 2],
  [4, 5, 6],
  [8, 9, 10]]
 

Klasy zagnieżdżone

klasa Opcje XlaConcatND Opcjonalne atrybuty dla XlaConcatND

Metody publiczne

Wyjście <T>
jako wyjście ()
Zwraca symboliczny uchwyt tensora.
statyczny <T> XlaConcatND <T>
utwórz (zakres zakresu , wejścia Iterable< Operand <T>>, List<Long> numConcats, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację XlaConcatND.
Wyjście <T>
statyczne opcje XlaConcatND
dopełnienia (List<Long> dopełnienia)

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 XlaConcatND <T> create (zakres zakresu , wejścia Iterable< Operand <T>>, List<Long> numConcats, Opcje... opcje)

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

Parametry
zakres aktualny zakres
wejścia Wprowadź wycinki tensora w kolejności od głównych wierszy, aby połączyć wszystkie wymiary. Wszystkie dane wejściowe muszą mieć ten sam kształt. } out_arg { nazwa: "wyjście" opis: <
liczbaPołączeń Liczba sposobów łączenia na wymiar.
opcje przenosi opcjonalne wartości atrybutów
Zwroty
  • nowa instancja XlaConcatND

publiczne wyjście <T> wyjście ()

publiczne statyczne dopełnienia XlaConcatND.Options (dopełnienia List<Long>)

Parametry
wyściółki Opcjonalna lista odpowiednich wypełnień na wymiar do usunięcia z końcowego scalonego tensora. Dopełnienia te nie mogą przekraczać rozmiaru wymiaru scalonego wyniku przed usunięciem dopełnienia.