T
- data type for outputValues()
output@Operator(group="sparse") public final class SparseSparseMaximum<T extends Number> extends PrimitiveOp
Assumes the two SparseTensors have the same shape, i.e., no broadcasting.
operation
Modifier and Type | Method and Description |
---|---|
static <T extends Number> |
create(Scope scope,
Operand<Long> aIndices,
Operand<T> aValues,
Operand<Long> aShape,
Operand<Long> bIndices,
Operand<T> bValues,
Operand<Long> bShape)
Factory method to create a class wrapping a new SparseSparseMaximum operation.
|
Output<Long> |
outputIndices()
2-D.
|
Output<T> |
outputValues()
1-D.
|
equals, hashCode, op, toString
public static <T extends Number> SparseSparseMaximum<T> create(Scope scope, Operand<Long> aIndices, Operand<T> aValues, Operand<Long> aShape, Operand<Long> bIndices, Operand<T> bValues, Operand<Long> bShape)
scope
- current scopeaIndices
- 2-D. `N x R` matrix with the indices of non-empty values in a
SparseTensor, in the canonical lexicographic ordering.aValues
- 1-D. `N` non-empty values corresponding to `a_indices`.aShape
- 1-D. Shape of the input SparseTensor.bIndices
- counterpart to `a_indices` for the other operand.bValues
- counterpart to `a_values` for the other operand; must be of the same dtype.bShape
- counterpart to `a_shape` for the other operand; the two shapes must be equal.Copyright © 2022. All rights reserved.