MatrixDiagPartV2

パブリック最終クラスMatrixDiagPartV2

バッチ化されたテンソルのバッチ化された対角部分を返します。

バッチ化された `input` の `k[0]` 番目から `k[1]` 番目の対角を持つテンソルを返します。

「input」が「r」次元「[I, J, ..., L, M, N]」を持つと仮定します。抽出するすべての対角線の最大長を `max_diag_len` とし、 `max_diag_len = min(M + min(k[1], 0), N + min(-k[0], 0))` とし、 `num_diags` とします。抽出する対角線の数、`num_diags = k[1] - k[0] + 1` になります。

`num_diags == 1` の場合、出力テンソルはランク `r - 1` で、形状 `[I, J, ..., L, max_diag_len]` と値:

diagonal[i, j, ..., l, n]
   = input[i, j, ..., l, n+y, n+x] ; if 0 <= n+y < M and 0 <= n+x < N,
     padding_value                 ; otherwise.
 
ここで、「y = max(-k[1], 0)」、「x = max(k[1], 0)」です。

それ以外の場合、出力テンソルはランク `r` を持ち、次元 `[I, J, ..., L, num_diags, max_diag_len]` の値を持ちます。

diagonal[i, j, ..., l, m, n]
   = input[i, j, ..., l, n+y, n+x] ; if 0 <= n+y < M and 0 <= n+x < N,
     padding_value                 ; otherwise.
 
ここで、「d = k[1] - m」、「y = max(-d, 0)」、および「x = max(d, 0)」です。

入力は少なくとも行列である必要があります。

例えば:

input = np.array([[[1, 2, 3, 4],  # Input shape: (2, 3, 4)
                    [5, 6, 7, 8],
                    [9, 8, 7, 6]],
                   [[5, 4, 3, 2],
                    [1, 2, 3, 4],
                    [5, 6, 7, 8]]])
 
 # A main diagonal from each batch.
 tf.matrix_diag_part(input) ==> [[1, 6, 7],  # Output shape: (2, 3)
                                 [5, 2, 7]]
 
 # A superdiagonal from each batch.
 tf.matrix_diag_part(input, k = 1)
   ==> [[2, 7, 6],  # Output shape: (2, 3)
        [4, 3, 8]]
 
 # A tridiagonal band from each batch.
 tf.matrix_diag_part(input, k = (-1, 1))
   ==> [[[2, 7, 6],  # Output shape: (2, 3, 3)
         [1, 6, 7],
         [5, 8, 0]],
        [[4, 3, 8],
         [5, 2, 7],
         [1, 6, 0]]]
 
 # Padding value = 9
 tf.matrix_diag_part(input, k = (1, 3), padding_value = 9)
   ==> [[[4, 9, 9],  # Output shape: (2, 3, 3)
         [3, 8, 9],
         [2, 7, 6]],
        [[2, 9, 9],
         [3, 4, 9],
         [4, 3, 8]]]
 

パブリックメソッド

出力<T>
asOutput ()
テンソルのシンボリック ハンドルを返します。
静的 <T> MatrixDiagPartV2 <T>
create (スコープscope、オペランド<T>入力、オペランド<整数>k、オペランド<T>paddingValue)
新しい MatrixDiagPartV2 操作をラップするクラスを作成するためのファクトリ メソッド。
出力<T>
対角線()
抽出された対角線。

継承されたメソッド

パブリックメソッド

public Output <T> asOutput ()

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

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

public static MatrixDiagPartV2 <T> create (スコープscope、オペランド<T>入力、オペランド<Integer> k、オペランド<T>paddingValue)

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

パラメータ
範囲現在のスコープ
入力`r >= 2` の場合、`r` テンソルをランク付けします。
k対角オフセット。正の値は上対角を意味し、0 は主対角を意味し、負の値は下対角を意味します。 「k」は、単一の整数 (単一の対角線の場合)、またはマトリックス バンドの下限と上限を指定する整数のペアにすることができます。 `k[0]` は `k[1]` より大きくてはなりません。
パディング値指定された対角バンドの外側の領域を埋める値。デフォルトは 0 です。
返品
  • MatrixDiagPartV2 の新しいインスタンス

public出力<T>対角線()

抽出された対角線。