텐서플로우:: 작전:: MatrixSetDiag

#include <array_ops.h>

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

요약

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 .

생성자와 소멸자

MatrixSetDiag (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input diagonal)

공개 속성

operation
output

공공 기능

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

공개 속성

작업

Operation operation

산출

::tensorflow::Output output

공공 기능

MatrixSetDiag

 MatrixSetDiag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input,
  ::tensorflow::Input diagonal
)

마디

::tensorflow::Node * node() const

연산자::텐서플로우::입력

 operator::tensorflow::Input() const

연산자::텐서플로우::출력

 operator::tensorflow::Output() const