tensorflow:: ops:: MatrixDiagPart

Devuelve la parte diagonal agrupada de un tensor agrupado.

Resumen

Esta operación devuelve un tensor con la parte diagonal de la input por lotes. La parte diagonal se calcula de la siguiente manera:

Supongamos que input tiene k dimensiones [I, J, K, ..., M, N] , entonces la salida es un tensor de rango k - 1 con dimensiones [I, J, K, ..., min(M, N)] donde:

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

La entrada debe ser al menos una matriz.

Por ejemplo:

# '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)

Argumentos:

• ámbito: Un objeto de ámbito
• Entrada: Tensor de rango k donde k >= 2 .

Devoluciones:

• Output : La(s) diagonal(es) extraída(s) con forma diagonal.shape = input.shape[:-2] + [min(input.shape[-2:])] .

Atributos públicos

::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