XlaSplitND
Zadbaj o dobrą organizację dzięki kolekcji
Zapisuj i kategoryzuj treści zgodnie ze swoimi preferencjami.
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]]
`liczba_podziałów`:
[2, 2]
i „wyściółki”:
[1, 1]
oczekiwane „produkty” to:
[[0, 1],
[3, 4]]
[[2, 0],
[5, 0]]
[[6, 7],
[0, 0]]
[[8, 0],
[0, 0]]
Metody dziedziczone
Z klasy java.lang.Object wartość logiczna | równa się (Obiekt arg0) |
ostatnia klasa<?> | pobierzKlasę () |
wew | hashCode () |
ostateczna pustka | powiadomić () |
ostateczna pustka | powiadom wszystkich () |
Smyczkowy | doString () |
ostateczna pustka | czekaj (długi arg0, int arg1) |
ostateczna pustka | czekaj (długi arg0) |
ostateczna pustka | Czekać () |
Z interfejsu java.lang.Iterable próżnia | forEach (Konsument<? super T> arg0) |
abstrakcyjny Iterator< Operand <T>> | iterator () |
Rozdzielacz< Operand <T>> | rozdzielacz () |
Metody publiczne
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
publiczny Iterator< Operand <T>> iterator ()
publiczna Lista< Wyjście <T>> wyjścia ()
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. |
---|
O ile nie stwierdzono inaczej, treść tej strony jest objęta licencją Creative Commons – uznanie autorstwa 4.0, a fragmenty kodu są dostępne na licencji Apache 2.0. Szczegółowe informacje na ten temat zawierają zasady dotyczące witryny Google Developers. Java jest zastrzeżonym znakiem towarowym firmy Oracle i jej podmiotów stowarzyszonych. Część treści jest objęta licencją numpy.
Ostatnia aktualizacja: 2024-08-24 UTC.
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Brak potrzebnych mi informacji"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Zbyt skomplikowane / zbyt wiele czynności do wykonania"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Nieaktualne treści"
},{
"type": "thumb-down",
"id": "translationIssue",
"label":"Problem z tłumaczeniem"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Problem z przykładami/kodem"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Inne"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Łatwo zrozumieć"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Rozwiązało to mój problem"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Inne"
}]
{
"lastModified": "Ostatnia aktualizacja: 2024-08-24 UTC.",
"confidential": False
}