텐서플로우:: 작전:: 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 .

인수:

• 범위: 범위 개체
• 입력: 순위 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