tf.raw_ops.SparseAccumulatorApplyGradient | TensorFlow v2.16.1 (original) (raw)
tf.raw_ops.SparseAccumulatorApplyGradient
Stay organized with collections Save and categorize content based on your preferences.
Applies a sparse gradient to a given accumulator.
View aliases
Compat aliases for migration
SeeMigration guide for more details.
tf.compat.v1.raw_ops.SparseAccumulatorApplyGradient
tf.raw_ops.SparseAccumulatorApplyGradient(
handle,
local_step,
gradient_indices,
gradient_values,
gradient_shape,
has_known_shape,
name=None
)
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, qint16, quint16, 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. |
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. Some content is licensed under the numpy license.
Last updated 2024-04-26 UTC.