tensorflow:: ops:: MatrixSetDiag

新しいバッチ化された対角値を持つバッチ化された行列テンソルを返します。

まとめ

inputとdiagonalが与えられた場合、この操作は、最も内側の行列の主対角成分を除いて、 inputと同じ形状と値を持つテンソルを返します。主対角成分は、 diagonalの値で上書きされます。

出力は以下のように計算されます。

input k+1次元[I, J, K, ..., M, N]を持ち、 diagonal成分がk次元[I, J, K, ..., min(M, N)]を持つと仮定します。このとき、出力は次元[I, J, K, ..., M, N]のランクk+1のテンソルとなります。

• output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] for m == n .
• output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] for m != n .

議論:

• スコープ:スコープオブジェクト
• 入力: ランクk+1 、ただしk >= 1 。
• 対角線: ランクk 、ただしk >= 1 。

返品:

• Output : ランクk+1 、 output.shape = input.shape 。

公開属性

Operation operation

::tensorflow::Output output

 MatrixSetDiag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input diagonal
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const