tf.assign_add
Update ref by adding value to it.
tf.assign_add(
ref, value, use_locking=None, name=None
)
This operation outputs "ref" after the update is done.
This makes it easier to chain operations that need to use the reset value.
Unlike tf.math.add, this op does not broadcast. ref and value must have
the same shape.
Args |
|---|
ref
|
A mutable Tensor. Must be one of the following types: float32,
float64, int64, int32, uint8, uint16, int16, int8,
complex64, complex128, qint8, quint8, qint32, half. Should be
from a Variable node.
|
value
|
A Tensor. Must have the same shape and dtype as ref. The value to
be added to the variable.
|
use_locking
|
An optional bool. Defaults to False. If True, the addition
will be protected by a lock; otherwise the behavior is undefined, but may
exhibit less contention.
|
name
|
A name for the operation (optional).
|
Returns |
|---|
|
Same as "ref". Returned as a convenience for operations that want
to use the new value after the variable has been updated.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2020-10-01 UTC.
[null,null,["Last updated 2020-10-01 UTC."],[],[]]
View source on GitHub