Zwraca część diagonalną tensora.
Ta operacja zwraca tensor z „przekątną” częścią „wejściowego”. Część „przekątną” oblicza się w następujący sposób:
Załóżmy, że „wejście” ma wymiary „[D1,..., Dk, D1,..., Dk]”, wówczas wyjściem jest tensor rzędu „k” o wymiarach „[D1,..., Dk]” Gdzie:
`przekątna[i1,..., ik] = wejście[i1, ..., ik, i1,..., ik]`.
Na przykład:
# 'input' is [[1, 0, 0, 0]
[0, 2, 0, 0]
[0, 0, 3, 0]
[0, 0, 0, 4]]
tf.diag_part(input) ==> [1, 2, 3, 4]
Stałe
Smyczkowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
Wyjście <T> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
statyczny <T rozszerza TType > TensorDiagPart <T> | |
Wyjście <T> | przekątna () Wyodrębniona przekątna. |
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.
public static TensorDiagPart <T> utwórz (zakres zakresu , wejście argumentu <T>)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację TensorDiagPart.
Parametry
zakres | aktualny zakres |
---|---|
wejście | Tensor rangi k, gdzie k jest parzyste, a nie zerowe. |
Powroty
- nowa instancja TensorDiagPart