tensorflow :: ops :: Diag

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 todos os outros lugares.

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 Scope
• diagonal: Tensor do Rank k onde k é no máximo 1.

Retorna:

• Output : o tensor de saída.

Atributos públicos

Operation operation

::tensorflow::Output output

 Diag(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input diagonal
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const