fluxo tensor:: ops:: ResourceScatterNdAdd
#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], use_resource=True) 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 identificador de recurso. Deve ser de um VarHandleOp.
- í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 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:
- a
Operation
criada
Construtores e Destruidores | |
---|---|
ResourceScatterNdAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates) | |
ResourceScatterNdAdd (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ResourceScatterNdAdd::Attrs & attrs) |
Atributos públicos | |
---|---|
operation |
Funções públicas | |
---|---|
operator::tensorflow::Operation () const |
Funções estáticas públicas | |
---|---|
UseLocking (bool x) |
Estruturas | |
---|---|
tensorflow:: ops:: ResourceScatterNdAdd:: Attrs | Configuradores de atributos opcionais para ResourceScatterNdAdd . |
Atributos públicos
operação
Operation operation
Funções públicas
ResourceScatterNdAdd
ResourceScatterNdAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
ResourceScatterNdAdd
ResourceScatterNdAdd( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ResourceScatterNdAdd::Attrs & attrs )
operador::tensorflow::Operação
operator::tensorflow::Operation() const
Funções estáticas públicas
UseLocking
Attrs UseLocking( bool x )