tf.raw_ops.MatrixSetDiag

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

tf.raw_ops.MatrixSetDiag(
    input, diagonal, name=None
)

Для заданных 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.