#include <state_ops.h>
Multiplies sparse updates into a variable reference.
This operation computes
# Scalar indices ref[indices, ...] *= updates[...]
# Vector indices (for each i) ref[indices[i], ...] *= updates[i, ...]
# High rank indices (for each i, ..., j) ref[indices[i, ..., j], ...] *= updates[i, ..., j, ...]
This operation outputs ref after the update is done. This makes it easier to chain operations that need to use the reset value.
Duplicate entries are handled correctly: if multiple indices reference the same location, their contributions multiply.
Requires updates.shape = indices.shape + ref.shape[1:] or updates.shape = [].
Arguments:
Variable node.ref.ref.Optional attributes (see Attrs):
Returns:
Output: = Same as ref. Returned as a convenience for operations that want to use the updated values after the update is done. | Constructors and Destructors | |
|---|---|
ScatterMul(const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates) | |
ScatterMul(const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ScatterMul::Attrs & attrs) |
| Public attributes | |
|---|---|
output_ref | |
| Public functions | |
|---|---|
node() const | ::tensorflow::Node * |
operator::tensorflow::Input() const | |
operator::tensorflow::Output() const | |
| Public static functions | |
|---|---|
UseLocking(bool x) | |
| Structs | |
|---|---|
| tensorflow::ops::ScatterMul::Attrs | Optional attribute setters for ScatterMul. |
::tensorflow::Output output_ref
ScatterMul( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates )
ScatterMul( const ::tensorflow::Scope & scope, ::tensorflow::Input ref, ::tensorflow::Input indices, ::tensorflow::Input updates, const ScatterMul::Attrs & attrs )
::tensorflow::Node * node() const
operator::tensorflow::Input() const
operator::tensorflow::Output() const
Attrs UseLocking( bool x )
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/scatter-mul.html