Aplica adição esparsa a valores individuais ou fatias em uma variável.
`ref` é um` Tensor` com classificação `P` e` índices` é um `Tensor` de classificação` Q`.
`índices` deve ser tensor inteiro, contendo índices em` ref`. Deve ser a forma `[d_0, ..., d_ {Q-2}, K]` onde `0 <K <= P`.
A dimensão mais interna de `índices` (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 espalhados 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]
Veja `tf.scatter_nd` para mais detalhes sobre como fazer atualizações em fatias.
Classes aninhadas
aula | ResourceScatterNdAdd.Options | Atributos opcionais para ResourceScatterNdAdd |
Métodos Públicos
static <T extends Number, U> ResourceScatterNdAdd | |
static ResourceScatterNdAdd.Options | useLocking (Boolean useLocking) |
Métodos herdados
Métodos Públicos
public static ResourceScatterNdAdd create ( Escopo do escopo, Operand <?> ref, Operand <T> índices, Operand <U> updates, Options ... options)
Método de fábrica para criar uma classe que envolve uma nova operação ResourceScatterNdAdd.
Parâmetros
alcance | escopo atual |
---|---|
ref | Um identificador de recurso. Deve ser de um VarHandleOp. |
índices | A Tensor. Deve ser um dos seguintes tipos: int32, int64. Um tensor de índices em ref. |
atualizações | A Tensor. Deve ter o mesmo tipo que ref. Um tensor de valores para adicionar à ref. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de ResourceScatterNdAdd
public static ResourceScatterNdAdd.Options useLocking (Boolean useLocking)
Parâmetros
useLocking | 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. |
---|