Łą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 „wejściowym”:
[[0, 1],
[4, 5]]
[[2, 3],
[6, 7]]
[[8, 9],
[12, 13]]
[[10, 11],
[14, 15]]
`liczba_podziałów`: [2, 2]
i „wyściółki”: [1, 1]
oczekiwane „produkty” 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> | |
Wyjście <T> | wyjście () |
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 |
Powroty
- nowa instancja XlaConcatND
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. |
---|