Dzieli tensor wejściowy zmiennej zasobu na wszystkie wymiary.
Operacja, która dzieli tensor wejściowy zmiennej zasobu na podstawie podanego atrybutu num_splits, opcjonalnie dopełnia plasterki i zwraca wycinki. Plasterki są zwracane w kolejności od głównego wiersza.
Ta operacja może zostać wygenerowana poprzez most TPU.
Na przykład, z tensorem `input`:
[[0, 1, 2],
[3, 4, 5],
[6, 7, 8]]
`num_splits`: [2, 2]
i `paddings`: [1, 1]
oczekiwane `outputs` to: [[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Klasy zagnieżdżone
klasa | Opcje odczytuVariableXlaSplitND | Opcjonalne atrybuty dla ReadVariableXlaSplitND |
Metody publiczne
statyczny <T> ReadVariableXlaSplitND <T> | utwórz (zakres zakresu , zasób argumentu <?>, klasa<T> T, długie N, lista<Long> numSplits, opcje... opcje) Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ReadVariableXlaSplitND. |
Iterator< Operand <T>> | iterator () |
Lista< Wyjście <T>> | wyjścia () |
statyczne ReadVariableXlaSplitND.Options | dopełnienia (List<Long> dopełnienia) |
Metody dziedziczone
Metody publiczne
public static ReadVariableXlaSplitND <T> create ( Zakres zakresu, Operand <?> zasób, Klasa<T> T, Długie N, List<Long> numSplits, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ReadVariableXlaSplitND.
Parametry
zakres | aktualny zakres |
---|---|
ratunek | Zmienna zasobu tensora wejściowego 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 |
Zwroty
- nowa instancja ReadVariableXlaSplitND
publiczne statyczne dopełnienia ReadVariableXlaSplitND.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. |
---|