tensorflow :: ops :: DiagPart
#include <array_ops.h>
テンソルの対角部分を返します。
概要
この操作は、 input
diagonal
部分を持つテンソルを返しinput
。 diagonal
部分は次のように計算されます。
input
次元が[D1,..., Dk, D1,..., Dk]
であるとすると、出力は次元[D1,..., Dk]
のランクk
テンソルになり[D1,..., Dk]
。
diagonal[i1,..., ik] = input[i1, ..., ik, i1,..., ik]
。
例えば:
# '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]
引数:
- スコープ:スコープオブジェクト
- 入力:kが偶数でゼロではないランクkテンソル。
戻り値:
-
Output
:抽出された対角線。
コンストラクタとデストラクタ | |
---|---|
DiagPart (const :: tensorflow::Scope & scope, :: tensorflow::Input input) |
パブリック属性 | |
---|---|
diagonal | |
operation |
公の行事 | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
パブリック属性
対角線
::tensorflow::Output diagonal
操作
Operation operation
公の行事
DiagPart
DiagPart( const ::tensorflow::Scope & scope, ::tensorflow::Input input )
ノード
::tensorflow::Node * node() const
operator :: tensorflow :: Input
operator::tensorflow::Input() const
operator :: tensorflow :: Output
operator::tensorflow::Output() const