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