przepływ tensorowy:: ops:: MatrixSetDiag

#include <array_ops.h>

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

Streszczenie

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

Dane wyjściowe 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)] . Następnie wynikiem jest tensor 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 Scope
  • dane wejściowe: Ranga k+1 , gdzie k >= 1 .
  • przekątna: Stopień k , gdzie k >= 1 .

Zwroty:

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

Konstruktory i destruktory

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

Atrybuty publiczne

operation
output

Funkcje publiczne

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

Atrybuty publiczne

operacja

Operation operation

wyjście

::tensorflow::Output output

Funkcje publiczne

MatrixSetDiag

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

węzeł

::tensorflow::Node * node() const 

operator::tensorflow::Wejście

 operator::tensorflow::Input() const 

operator::tensorflow::Wyjście

 operator::tensorflow::Output() const