tensorflow :: ops :: ScatterNdAdd
#include <state_ops.h>
Aplica adição esparsa a valores individuais ou fatias em uma variável .
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 forma [d_0, ..., d_{Q-2}, K]
onde 0 < K <= P
A dimensão mais interna dos indices
(com comprimento K
) corresponde aos índices em elementos (se K = P
) ou fatias (se K < P
) ao longo da K
ésima dimensão de ref
.
updates
é o 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 em fatias.
Argumentos:
- escopo: um objeto Scope
- 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 em ref.
- atualizações: um tensor . Deve ter o mesmo tipo do ref. Um tensor de valores atualizados para adicionar à ref.
Atributos opcionais (consulte Attrs
):
- use_locking: um bool opcional. O padrão é True. Se for True, a atribuição será protegida por um cadeado; caso contrário, o comportamento é indefinido, mas pode exibir 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) |
Structs | |
---|---|
tensorflow :: ops :: ScatterNdAdd :: Attrs | Configuradores de atributos opcionais para ScatterNdAdd . |
Atributos públicos
Operação
Operation operation
output_ref
::tensorflow::Output output_ref
Funções públicas
ScatterNdAdd
ScatterNdAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
ScatterNdAdd
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 )