tensorflow::ops::MatrixSetDiag

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

Краткое содержание

При наличии 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] for m == n .
• output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] for m != n .

Аргументы:

• scope: Объект Scope
• Входные данные: ранг 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