tensorflow :: operaciones :: ResourceScatterNdSub
#include <state_ops.h>
Aplica una resta dispersa a valores individuales o porciones en una variable .
Resumen
ref
es un Tensor
con rango P
e indices
es un Tensor
de rango Q
indices
deben ser un tensor entero, que contenga índices en ref
. Debe tener la forma [d_0, ..., d_{Q-2}, K]
donde 0 < K <= P
La dimensión más interna de los indices
(con longitud K
) corresponde a índices en elementos (si K = P
) o rebanadas (si K < P
) a lo largo de la K
ésima dimensión de la ref
.
updates
es Tensor
de rango Q-1+PK
con forma:
[d_0, ..., d_{Q-2}, ref.shape[K], ..., ref.shape[P-1]]
Por ejemplo, digamos que queremos restar 4 elementos dispersos de un tensor de rango 1 con 8 elementos. En Python, esa resta se vería así:
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]) sub = tf.scatter_nd_sub(ref, indices, updates) with tf.Session() as sess: print sess.run(sub)
La actualización resultante para ref se vería así:
[1, -9, 3, -6, -4, 6, 7, -4]
Consulte tf.scatter_nd
para obtener más detalles sobre cómo realizar actualizaciones en los segmentos.
Argumentos:
- alcance: un objeto de alcance
- ref: un identificador de recursos. Debe ser de VarHandleOp.
- índices: un tensor . Debe ser uno de los siguientes tipos: int32, int64. Un tensor de índices en ref.
- actualizaciones: un tensor . Debe tener el mismo tipo que la ref. Un tensor de valores para agregar a la ref.
Atributos opcionales (consulte Attrs
):
- use_locking: un bool opcional. El valor predeterminado es Verdadero. Si es Verdadero, la asignación estará protegida por un candado; de lo contrario, el comportamiento no está definido, pero puede mostrar menos contención.
Devoluciones:
- la
Operation
creada
Constructores y Destructores | |
---|---|
ResourceScatterNdSub (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates) | |
ResourceScatterNdSub (const :: tensorflow::Scope & scope, :: tensorflow::Input ref, :: tensorflow::Input indices, :: tensorflow::Input updates, const ResourceScatterNdSub::Attrs & attrs) |
Atributos públicos | |
---|---|
operation |
Funciones publicas | |
---|---|
operator::tensorflow::Operation () const |
Funciones estáticas públicas | |
---|---|
UseLocking (bool x) |
Estructuras | |
---|---|
tensorflow :: ops :: ResourceScatterNdSub :: Attrs | Establecedores de atributos opcionales para ResourceScatterNdSub . |
Atributos públicos
operación
Operation operation
Funciones publicas
ResourceScatterNdSub
ResourceScatterNdSub( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
ResourceScatterNdSub
ResourceScatterNdSub( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ResourceScatterNdSub::Attrs & attrs )
operador :: tensorflow :: Operación
operator::tensorflow::Operation() const
Funciones estáticas públicas
UseLocking
Attrs UseLocking( bool x )