тензорный поток:: опс:: 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] для m == n .
• output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] для m != n .

Аргументы:

• область: объект области.
• ввод: Rank 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