Google I/O에 참여해 주셔서 감사합니다. 주문형 모든 세션 보기 주문형 시청

tensorflow :: ops :: MatrixSetDiag

#include <array_ops.h>

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

요약

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

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

inputk+1 차원 [I, J, K, ..., M, N] 이고 diagonalk 차원 [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] m != n 대한 output[i, j, k, ..., m, n] = input[i, j, k, ..., 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 

연산자 :: tensorflow :: 입력

 operator::tensorflow::Input() const 

연산자 :: tensorflow :: 출력

 operator::tensorflow::Output() const