tensorflow::ops::ScatterNdUpdate

#include <state_ops.h>

Aplica updates esparsas a valores individuais ou fatias dentro de um determinado conjunto.

Resumo

variável de acordo com indices .

ref é um Tensor de classificação P e indices é um Tensor de classificação Q

indices devem ser tensores inteiros, contendo os índices de ref . Devem ter o formato \([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 é Tensor de classificação Q-1+PK com a seguinte forma:

$$[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]].$$

Por exemplo, digamos que queremos atualizar um tensor de posto 1 com 8 elementos, composto por 4 elementos dispersos. Em Python, essa atualização seria feita da seguinte forma:

    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 da referência ficaria assim:

[1, 11, 3, 10, 9, 6, 7, 12]

Consulte tf.scatter_nd para obter mais detalhes sobre como atualizar as fatias.

Veja também tf.scatter_update e tf.batch_scatter_update .

Argumentos:

  • escopo: Um objeto de escopo
  • ref: Um Tensor mutável. Deve vir de um nó de Variável .
  • índices: Um Tensor . Deve ser um dos seguintes tipos: int32, int64. Um tensor de índices em ref.
  • updates: Um Tensor . Deve ter o mesmo tipo que ref. Um tensor de valores atualizados para adicionar a ref.

Atributos opcionais (ver Attrs ):

  • use_locking: Um booleano opcional. O valor padrão é True. Se True, a atribuição será protegida por um bloqueio; caso contrário, o comportamento é indefinido, mas pode apresentar menos contenção.

Devoluções:

  • Output : Igual à referência. Retornado como uma conveniência para operações que desejam usar os valores atualizados após a conclusão da atualização.

Construtores e Destrutores

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

Definidores de atributos opcionais para ScatterNdUpdate .

Atributos públicos

operação

Operation operation

referência_de_saída

::tensorflow::Output output_ref

Funções públicas

Atualização de dispersão

 ScatterNdUpdate(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input ref,
  ::tensorflow::Input indices,
  ::tensorflow::Input updates
)

Atualização de dispersão

 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

UsarBloqueio

Attrs UseLocking(
  bool x
)