Łą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 zapisuje połączony tensor bez dopełnień w zmiennej zasobu.
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]]
[2, 2]
[1, 1]
[[0, 1, 2],
[4, 5, 6],
[8, 9, 10]]
Klasy zagnieżdżone
klasa | Przypisz zmiennąXlaConcatND.Opcje | Opcjonalne atrybuty dla AssignVariableXlaConcatND |
Metody publiczne
statyczny <T> Przypisz zmiennąXlaConcatND | utwórz (zakres zakresu , zasób argumentu <?>, wejścia Iterable< argumentu <T>>, lista <Long> numConcats, opcje... opcje) Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację AssignVariableXlaConcatND. |
statyczne AssignVariableXlaConcatND.Options | dopełnienia (List<Long> dopełnienia) |
Metody dziedziczone
Metody publiczne
public static AssignVariableXlaConcatND create (zakres zakresu , zasób argumentu <?>, wejścia Iterowalne< operand <T>>, lista<Long> numConcats, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację AssignVariableXlaConcatND.
Parametry
zakres | aktualny zakres |
---|---|
ratunek | Zmienna zasobu dla połączonych tensorów wejściowych we wszystkich wymiarach. } in_arg { nazwa: "wejścia" opis: < |
liczbaPołączeń | Liczba sposobów łączenia na wymiar. |
opcje | przenosi opcjonalne wartości atrybutów |
Powroty
- nowa instancja AssignVariableXlaConcatND
publiczne statyczne dopełnienia AssignVariableXlaConcatND.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. |
---|