fluxo tensor:: ops:: DispersãoNdAdicionar
#include <state_ops.h>
Aplica adição esparsa a valores individuais ou fatias em uma Variable .
Resumo
ref
é um Tensor
com classificação P
e indices
é um Tensor
de classificação Q
.
indices
devem ser tensores inteiros, contendo índices em ref
. Deve ser a forma [d_0, ..., d_{Q-2}, K]
onde 0 < K <= P
.
A dimensão mais interna dos indices
(com comprimento K
) corresponde a índices em elementos (se K = P
) ou fatias (se K < P
) ao longo da K
-ésima dimensão de ref
.
updates
é Tensor
de classificação Q-1+PK
com forma:
[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]]
Por exemplo, digamos que queremos adicionar 4 elementos dispersos a um tensor de classificação 1 para 8 elementos. Em Python, essa adição ficaria assim:
ref = tf.Variable([1, 2, 3, 4, 5, 6, 7, 8]) indices = tf.constant([[4], [3], [1], [7]]) updates = tf.constant([9, 10, 11, 12]) add = tf.scatter_nd_add(ref, indices, updates) with tf.Session() as sess: print sess.run(add)
A atualização resultante para ref ficaria assim:
[1, 13, 3, 14, 14, 6, 7, 20]
Consulte tf.scatter_nd
para obter mais detalhes sobre como fazer atualizações nas fatias.
Argumentos:
- escopo: um objeto Escopo
- ref: Um tensor mutável. Deve ser de um nó Variável .
- índices: Um tensor . Deve ser um dos seguintes tipos: int32, int64. Um tensor de índices na ref.
- atualizações: Um Tensor . Deve ter o mesmo tipo da ref. Um tensor de valores atualizados para adicionar à ref.
Atributos opcionais (veja Attrs
):
- use_locking: Um bool opcional. O padrão é Verdadeiro. Se for True, a atribuição será protegida por um bloqueio; caso contrário, o comportamento será indefinido, mas poderá apresentar menos contenção.
Retorna:
-
Output
: Igual à ref. Retornado como uma conveniência para operações que desejam usar os valores atualizados após a atualização ser concluída.
Construtores e Destruidores | |
---|---|
ScatterNdAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates) | |
ScatterNdAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ScatterNdAdd::Attrs & attrs) |
Atributos públicos | |
---|---|
operation | |
output_ref |
Funções públicas | |
---|---|
node () const | ::tensorflow::Node * |
operator::tensorflow::Input () const | |
operator::tensorflow::Output () const |
Funções estáticas públicas | |
---|---|
UseLocking (bool x) |
Estruturas | |
---|---|
tensorflow:: ops:: ScatterNdAdd:: Attrs | Configuradores de atributos opcionais para ScatterNdAdd . |
Atributos públicos
operação
Operation operation
saída_ref
::tensorflow::Output output_ref
Funções públicas
DispersãoNdAdicionar
ScatterNdAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
DispersãoNdAdicionar
ScatterNdAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ScatterNdAdd::Attrs & attrs )
nó
::tensorflow::Node * node() const
operador::tensorflow::Input
operator::tensorflow::Input() const
operador::tensorflow::Saída
operator::tensorflow::Output() const
Funções estáticas públicas
UseLocking
Attrs UseLocking( bool x )