W3cubDocs

/TensorFlow 2.4

tf.raw_ops.SparseAccumulatorApplyGradient

Applies a sparse gradient to a given accumulator.

Does not add if local_step is smaller than the accumulator's global_step.

Args
handle A Tensor of type mutable string. The handle to a accumulator.
local_step A Tensor of type int64. The local_step value at which the sparse gradient was computed.
gradient_indices A Tensor of type int64. Indices of the sparse gradient to be accumulated. Must be a vector.
gradient_values A Tensor. Must be one of the following types: float32, float64, int32, uint8, int16, int8, complex64, int64, qint8, quint8, qint32, bfloat16, uint16, complex128, half, uint32, uint64. Values are the non-zero slices of the gradient, and must have the same first dimension as indices, i.e., the nnz represented by indices and values must be consistent.
gradient_shape A Tensor of type int64. Shape of the sparse gradient to be accumulated.
has_known_shape A bool. Boolean indicating whether gradient_shape is unknown, in which case the input is ignored during validation.
name A name for the operation (optional).
Returns
The created Operation.

© 2020 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/versions/r2.4/api_docs/python/tf/raw_ops/SparseAccumulatorApplyGradient