テンソルフロー::作戦::マトリックスセットダイアグ

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

まとめ

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のテンソルになります。

• m == nの場合output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] 。
• output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] ( 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