tensorflow:: ops:: MatrixSetDiag

Zwraca tensor macierzy wsadowej z nowymi wsadowymi wartościami przekątnej.

Streszczenie

Biorąc pod uwagę input i diagonal , ta operacja zwraca tensor o tym samym kształcie i wartościach co input , z wyjątkiem głównej przekątnej macierzy najbardziej wewnętrznych. Zostaną one nadpisane wartościami w diagonal .

Wynik oblicza się w następujący sposób:

Załóżmy, że input mają k+1 wymiarów [I, J, K, ..., M, N] , a diagonal ma k wymiarów [I, J, K, ..., min(M, N)] . Wówczas dane wyjściowe są tensorem rzędu k+1 o wymiarach [I, J, K, ..., M, N] , gdzie:

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

Argumenty:

• zakres: obiekt zakresu
• wejście: Ranga k+1 , gdzie k >= 1 .
• przekątna: Rząd k , gdzie k >= 1 .

Zwroty:

• Output : Ranga k+1 , przy czym output.shape = input.shape .

Atrybuty publiczne

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