텐서플로우:: ops:: MatrixSetDiag

새로운 대각선 값을 포함하는 배치 행렬 텐서를 반환합니다.

요약

input 과 diagonal 주어졌을 때, 이 연산은 가장 안쪽 행렬의 주대각선을 제외하고 input 과 동일한 모양과 값을 가진 텐서를 반환합니다. 가장 안쪽 행렬의 주대각선은 diagonal 요소의 값으로 덮어쓰여집니다.

출력값은 다음과 같이 계산됩니다.

input k+1 차원 [I, J, K, ..., M, N] 이고 diagonal 가 k 차원 [I, J, K, ..., min(M, N)] 이라고 가정합니다. 그러면 출력은 차원이 [I, J, K, ..., M, N] 인 k+1 랭크 텐서입니다. 여기서:

• output[i, j, k, ..., m, n] = diagonal[i, j, k, ..., n] for m == n .
• output[i, j, k, ..., m, n] = input[i, j, k, ..., m, n] for m != n .

논거:

• 범위: Scope 객체
• 입력: 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