ReadVariableXlaSplitND

publiczna klasa końcowa ReadVariableXlaSplitND

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>>
Lista< Wyjście <T>>
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

publiczny Iterator< Operand <T>> iterator ()

publiczna Lista< Wyjście <T>> wyjścia ()

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.