tensorflow::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
• input: ранг k+1, где k >= 1.
• diagonal: ранг 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