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

#include <array_ops.h>

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

まとめ

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 。

コンストラクターとデストラクター

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

パブリック属性

operation
output

公共機能

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

パブリック属性

手術

Operation operation

出力

::tensorflow::Output output

公共機能

マトリックスセットダイアグ

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

ノード

::tensorflow::Node * node() const

演算子::tensorflow::入力

 operator::tensorflow::Input() const

演算子::tensorflow::出力

 operator::tensorflow::Output() const