fluxo tensor:: ops:: ScatterNdUpdate
#include <state_ops.h> Aplica updates esparsas a valores individuais ou fatias dentro de um determinado.
Resumo
variável de acordo com indices .
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 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 atualizar 4 elementos dispersos para um tensor de classificação 1 para 8 elementos. Em Python, essa atualizaçã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]) update = tf.scatter_nd_update(ref, indices, updates) with tf.Session() as sess: print sess.run(update)
A atualização resultante para ref ficaria assim:
[1, 11, 3, 10, 9, 6, 7, 12]
Consulte tf.scatter_nd para obter mais detalhes sobre como fazer atualizações nas fatias.
Consulte também tf.scatter_update e tf.batch_scatter_update .
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 | |
|---|---|
ScatterNdUpdate (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates) | |
ScatterNdUpdate (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ScatterNdUpdate::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:: ScatterNdUpdate:: Attrs | Configuradores de atributos opcionais para ScatterNdUpdate . |
Atributos públicos
operação
Operation operation
saída_ref
::tensorflow::Output output_ref
Funções públicas
ScatterNdUpdate
ScatterNdUpdate( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
ScatterNdUpdate
ScatterNdUpdate( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ScatterNdUpdate::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 )