Dzieli tensor wejściowy na wszystkie wymiary.
Operacja, która dzieli tensor wejściowy na podstawie podanego atrybutu num_splits, opcjonalnie dopełnia plasterki i zwraca plasterki. Plasterki są zwracane w kolejności od głównego wiersza.
Ta operacja może zostać wygenerowana poprzez most TPU.
Na przykład z tensorem „wejściowym”:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
[2, 2]
[1, 1]
[[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Klasy zagnieżdżone
klasa | Opcje XlaSplitND | Opcjonalne atrybuty dla XlaSplitND |
Metody publiczne
statyczny <T> XlaSplitND <T> | |
Iterator< Operand <T>> | iterator () |
Lista< Wyjście <T>> | wyjścia () |
statyczne opcje XlaSplitND | dopełnienia (List<Long> dopełnienia) |
Metody dziedziczone
Metody publiczne
public static XlaSplitND <T> create (Zakres zakresu , Operand <T> wejście, Długie N, List<Long> numSplits, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację XlaSplitND.
Parametry
zakres | aktualny zakres |
---|---|
wejście | Tensor wejściowy do podziału na wszystkie wymiary. } out_arg { nazwa: "wyjścia" opis: < |
liczbapodziałów | Liczba sposobów podziału na wymiar. Wymiary kształtu muszą być równomiernie podzielne. |
opcje | przenosi opcjonalne wartości atrybutów |
Powroty
- nowa instancja XlaSplitND
publiczne statyczne dopełnienia XlaSplitND.Options (dopełnienia List<Long>)
Parametry
wyściółki | Opcjonalna lista odpowiednich wypełnień według wymiaru tensora wejściowego do zastosowania przed podziałem. Można to wykorzystać, aby wymiar był równomiernie podzielny. |
---|