tensoreflusso:: ops:: MatrixDiagPart

Restituisce la parte diagonale in batch di un tensore in batch.

Riepilogo

Questa operazione restituisce un tensore con la parte diagonal input batch. La parte diagonal si calcola come segue:

Supponiamo che input abbia k dimensioni [I, J, K, ..., M, N] , quindi l'output è un tensore di rango k - 1 con dimensioni [I, J, K, ..., min(M, N)] Dove:

diagonal[i, j, k, ..., n] = input[i, j, k, ..., n, n] .

L'input deve essere almeno una matrice.

Per esempio:

# 'input' is [[[1, 0, 0, 0]
               [0, 2, 0, 0]
               [0, 0, 3, 0]
               [0, 0, 0, 4]],
              [[5, 0, 0, 0]
               [0, 6, 0, 0]
               [0, 0, 7, 0]
               [0, 0, 0, 8]]]

and input.shape = (2, 4, 4)

tf.matrix_diag_part(input) ==> [[1, 2, 3, 4], [5, 6, 7, 8]]

which has shape (2, 4)

Argomenti:

• scope: un oggetto Scope
• input: tensore di rango k dove k >= 2 .

Resi:

• Output : le diagonali estratte aventi forma diagonal.shape = input.shape[:-2] + [min(input.shape[-2:])] .

Attributi pubblici

::tensorflow::Output diagonal

Operation operation

 MatrixDiagPart(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input input
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const