flujo tensor:: operaciones:: MatrixDiagPart

#include <array_ops.h>

Devuelve la parte diagonal por lotes de un tensor por lotes.

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)] dónde:

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:

  • alcance: un objeto de alcance
  • entrada: Rango k tensor donde k >= 2 .

Devoluciones:

  • Output : Las diagonales extraídas tienen forma diagonal.shape = input.shape[:-2] + [min(input.shape[-2:])] .

Constructores y destructores

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

Atributos públicos

diagonal
operation

Funciones públicas

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

Atributos públicos

diagonal 

::tensorflow::Output diagonal

operación 

Operation operation

Funciones públicas

MatrixDiagPart 

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

nodo 

::tensorflow::Node * node() const

operador::tensorflow::Entrada 

 operator::tensorflow::Input() const

operador::tensorflow::Salida 

 operator::tensorflow::Output() const