Skopiuj tensor, ustawiając wszystko poza środkowym pasmem w każdej najbardziej wewnętrznej macierzy na zero.
Część „pasmową” oblicza się w następujący sposób: Załóżmy, że „wejście” ma „k” wymiarów „[I, J, K, ..., M, N]”, wówczas wynikiem jest tensor o tym samym kształcie, gdzie
`pasmo[i, j, k, ..., m, n] = in_band(m, n) * wejście[i, j, k, ..., m, n]`.
Funkcja wskaźnika
`in_band(m, n) = (liczba_dolna < 0 || (mn) <= liczba_dolna)) && (liczba_górna < 0 || (nm) <= liczba_górna)`.
Na przykład:
# if 'input' is [[ 0, 1, 2, 3]
[-1, 0, 1, 2]
[-2, -1, 0, 1]
[-3, -2, -1, 0]],
tf.matrix_band_part(input, 1, -1) ==> [[ 0, 1, 2, 3]
[-1, 0, 1, 2]
[ 0, -1, 0, 1]
[ 0, 0, -1, 0]],
tf.matrix_band_part(input, 2, 1) ==> [[ 0, 1, 0, 0]
[-1, 0, 1, 0]
[-2, -1, 0, 1]
[ 0, -2, -1, 0]]
Przydatne przypadki specjalne: tf.matrix_band_part(input, 0, -1) ==> Upper triangular part.
tf.matrix_band_part(input, -1, 0) ==> Lower triangular part.
tf.matrix_band_part(input, 0, 0) ==> Diagonal.
Stałe
Smyczkowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
Wyjście <T> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
Wyjście <T> | zespół () Tensor rangi „k” o tym samym kształcie co dane wejściowe. |
statyczny <T rozszerza TType , U rozszerza TNumber > BandPart <T> |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
publiczne wyjście <T> asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
publiczne pasmo wyjściowe <T> ()
Tensor rangi „k” o tym samym kształcie co dane wejściowe. Wyodrębniony tensor pasmowy.
public static BandPart <T> utwórz (zakres zakresu , argument <T> wejście, argument <U> numLower, argument <U> numUpper)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację BandPart.
Parametry
zakres | aktualny zakres |
---|---|
wejście | Tensor rangi „k”. |
liczbaDolna | Tensor 0-D. Liczba podprzekątnych do zachowania. Jeżeli wynik jest ujemny, zachowaj cały dolny trójkąt. |
liczbaGórna | Tensor 0-D. Liczba superprzekątnych do zachowania. Jeśli wynik jest ujemny, zachowaj cały górny trójkąt. |
Powroty
- nowa instancja BandPart