fluxo tensor:: ops:: Diag.
#include <array_ops.h>
Retorna um tensor diagonal com determinados valores diagonais.
Resumo
Dada uma diagonal
, esta operação retorna um tensor com a diagonal
e todo o resto preenchido com zeros. A diagonal é calculada da seguinte forma:
Suponha que diagonal
tenha dimensões [D1,..., Dk], então a saída é um tensor de classificação 2k com dimensões [D1,..., Dk, D1,..., Dk] onde:
output[i1,..., ik, i1,..., ik] = diagonal[i1, ..., ik]
e 0 em qualquer outro lugar.
Por exemplo:
# 'diagonal' is [1, 2, 3, 4] tf.diag(diagonal) ==> [[1, 0, 0, 0] [0, 2, 0, 0] [0, 0, 3, 0] [0, 0, 0, 4]]
Argumentos:
- escopo: um objeto Escopo
- diagonal: Tensor de classificação k onde k é no máximo 1.
Retorna:
-
Output
: o tensor de saída.
Construtores e Destruidores | |
---|---|
Diag (const :: tensorflow::Scope & scope, :: tensorflow::Input diagonal) |
Atributos públicos | |
---|---|
operation | |
output |
Funções públicas | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Atributos públicos
operação
Operation operation
saída
::tensorflow::Output output
Funções públicas
Diag.
Diag( const ::tensorflow::Scope & scope, ::tensorflow::Input diagonal )
nó
::tensorflow::Node * node() const
operador::tensorflow::Input
operator::tensorflow::Input() const
operador::tensorflow::Saída
operator::tensorflow::Output() const