T
- data type for outputValues()
output@Operator(group="sparse") public final class SparseReduceSumSparse<T> extends PrimitiveOp
This Op takes a SparseTensor and is the sparse counterpart to `tf.reduce_sum()`. In contrast to SparseReduceSum, this Op returns a SparseTensor.
Reduces `sp_input` along the dimensions given in `reduction_axes`. Unless `keep_dims` is true, the rank of the tensor is reduced by 1 for each entry in `reduction_axes`. If `keep_dims` is true, the reduced dimensions are retained with length 1.
If `reduction_axes` has no entries, all dimensions are reduced, and a tensor with a single element is returned. Additionally, the axes can be negative, which are interpreted according to the indexing rules in Python.
Modifier and Type | Class and Description |
---|---|
static class |
SparseReduceSumSparse.Options
Optional attributes for
SparseReduceSumSparse |
operation
Modifier and Type | Method and Description |
---|---|
static <T> SparseReduceSumSparse<T> |
create(Scope scope,
Operand<Long> inputIndices,
Operand<T> inputValues,
Operand<Long> inputShape,
Operand<Integer> reductionAxes,
SparseReduceSumSparse.Options... options)
Factory method to create a class wrapping a new SparseReduceSumSparse operation.
|
static SparseReduceSumSparse.Options |
keepDims(Boolean keepDims) |
Output<Long> |
outputIndices() |
Output<Long> |
outputShape() |
Output<T> |
outputValues() |
equals, hashCode, op, toString
public static <T> SparseReduceSumSparse<T> create(Scope scope, Operand<Long> inputIndices, Operand<T> inputValues, Operand<Long> inputShape, Operand<Integer> reductionAxes, SparseReduceSumSparse.Options... options)
scope
- current scopeinputIndices
- 2-D. `N x R` matrix with the indices of non-empty values in a
SparseTensor, possibly not in canonical ordering.inputValues
- 1-D. `N` non-empty values corresponding to `input_indices`.inputShape
- 1-D. Shape of the input SparseTensor.reductionAxes
- 1-D. Length-`K` vector containing the reduction axes.options
- carries optional attributes valuespublic static SparseReduceSumSparse.Options keepDims(Boolean keepDims)
keepDims
- If true, retain reduced dimensions with length 1.Copyright © 2022. All rights reserved.