tensorflow::ops::MatrixSetDiag
tensorflow::ops::MatrixSetDiag
#include <array_ops.h>
Returns a batched matrix tensor with new batched diagonal values.
Summary
Given input
and diagonal
, this operation returns a tensor with the same shape and values as input
, except for the main diagonal of the innermost matrices. These will be overwritten by the values in diagonal
.
The output is computed as follows:
Assume input
has k+1
dimensions [I, J, K, ..., M, N]
and diagonal
has k
dimensions [I, J, K, ..., min(M, N)]
. Then the output is a tensor of rank k+1
with dimensions [I, J, K, ..., M, N]
where:
-
output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n]
form == n
. -
output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n]
for登录查看完整内容