AssignVariableXlaConcatND

publiczna klasa końcowa AssignVariableXlaConcatND

Łą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 `input`: 

[[0, 1],
  [4, 5]]
 [[2, 3],
  [6, 7]]
 [[8, 9],
  [12, 13]]
 [[10, 11],
  [14, 15]]
`num_splits`: 
[2, 2]
i `paddings`: 
[1, 1]
oczekiwane `outputs` to: 
[[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
Zwroty
  • 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.