tf.raw_ops.SparseAccumulatorApplyGradient
Applies a sparse gradient to a given accumulator.
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-01-23 UTC.
[null,null,["Last updated 2024-01-23 UTC."],[],[]]