тензорный поток :: ops :: MatrixSetDiag

#include <array_ops.h>

Возвращает пакетный матричный тензор с новыми пакетными диагональными значениями.

Резюме

Учитывая input и diagonal , эта операция возвращает тензор той же формы и значений, что и input , за исключением главной диагонали самых внутренних матриц. Они будут заменены значениями по diagonal .

Выходные данные рассчитываются следующим образом:

Предположим, что input имеет k+1 размерностей [I, J, K, ..., M, N] а diagonal имеет k размеров [I, J, K, ..., min(M, N)] . Тогда на выходе будет тензор ранга k+1 с размерностями [I, J, K, ..., M, N] где:

  • output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] для m == n .
  • output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] для m != n .

Аргументы:

  • scope: Объект Scope
  • ввод: Ранг 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

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

узел

::tensorflow::Node * node() const 

оператор :: тензор потока :: Вход

 operator::tensorflow::Input() const 

оператор :: тензор потока :: Вывод

 operator::tensorflow::Output() const