TensorDiagPart

パブリック最終クラスTensorDiagPart

テンソルの対角部分を返します。

この演算は、「input」の「diagonal」部分を含むテンソルを返します。 「対角」部分は次のように計算されます。

「入力」の次元が「[D1,..., Dk, D1,..., Dk]」であると仮定すると、出力は次元「[D1,..., Dk]」を持つランク「k」のテンソルになります。どこ:

`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]
 

定数

OP_NAME TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
static <T extends TType > TensorDiagPart <T>
create (スコープscope、オペランド<T>入力)
新しい TensorDiagPart オペレーションをラップするクラスを作成するためのファクトリ メソッド。
出力<T>
対角線()
抽出された対角線。

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: "DiagPart"

パブリックメソッド

public Output <T> asOutput ()

テンソルのシンボリック ハンドルを返します。

TensorFlow オペレーションへの入力は、別の TensorFlow オペレーションの出力です。このメソッドは、入力の計算を表すシンボリック ハンドルを取得するために使用されます。

public static TensorDiagPart <T> create (スコープスコープ、オペランド<T> 入力)

新しい TensorDiagPart オペレーションをラップするクラスを作成するファクトリ メソッド。

パラメータ
範囲現在のスコープ
入力ランク k テンソル。k は偶数でゼロではありません。
返品
  • TensorDiagPart の新しいインスタンス

public出力<T>対角線()

抽出された対角線。