T
- data type for outputValues()
output@Operator(group="sparse") public final class SparseCross<T> extends PrimitiveOp
The op takes two lists, one of 2D `SparseTensor` and one of 2D `Tensor`, each representing features of one feature column. It outputs a 2D `SparseTensor` with the batchwise crosses of these features.
For example, if the inputs are
inputs[0]: SparseTensor with shape = [2, 2] [0, 0]: "a" [1, 0]: "b" [1, 1]: "c"
inputs[1]: SparseTensor with shape = [2, 1] [0, 0]: "d" [1, 0]: "e"
inputs[2]: Tensor [["f"], ["g"]]
then the output will be
shape = [2, 2] [0, 0]: "a_X_d_X_f" [1, 0]: "b_X_e_X_g" [1, 1]: "c_X_e_X_g"
if hashed_output=true then the output will be
shape = [2, 2] [0, 0]: FingerprintCat64( Fingerprint64("f"), FingerprintCat64( Fingerprint64("d"), Fingerprint64("a"))) [1, 0]: FingerprintCat64( Fingerprint64("g"), FingerprintCat64( Fingerprint64("e"), Fingerprint64("b"))) [1, 1]: FingerprintCat64( Fingerprint64("g"), FingerprintCat64( Fingerprint64("e"), Fingerprint64("c")))
operation
Modifier and Type | Method and Description |
---|---|
static <T,U> SparseCross<T> |
create(Scope scope,
Iterable<Operand<Long>> indices,
Iterable<Operand<?>> values,
Iterable<Operand<Long>> shapes,
Iterable<Operand<?>> denseInputs,
Boolean hashedOutput,
Long numBuckets,
Long hashKey,
Class<T> outType,
Class<U> internalType)
Factory method to create a class wrapping a new SparseCross operation.
|
Output<Long> |
outputIndices()
2-D.
|
Output<Long> |
outputShape()
1-D.
|
Output<T> |
outputValues()
1-D.
|
equals, hashCode, op, toString
public static <T,U> SparseCross<T> create(Scope scope, Iterable<Operand<Long>> indices, Iterable<Operand<?>> values, Iterable<Operand<Long>> shapes, Iterable<Operand<?>> denseInputs, Boolean hashedOutput, Long numBuckets, Long hashKey, Class<T> outType, Class<U> internalType)
scope
- current scopeindices
- 2-D. Indices of each input `SparseTensor`.values
- 1-D. values of each `SparseTensor`.shapes
- 1-D. Shapes of each `SparseTensor`.denseInputs
- 2-D. Columns represented by dense `Tensor`.hashedOutput
- If true, returns the hash of the cross instead of the string.
This will allow us avoiding string manipulations.numBuckets
- It is used if hashed_output is true.
output = hashed_value%num_buckets if num_buckets > 0 else hashed_value.hashKey
- Specify the hash_key that will be used by the `FingerprintCat64`
function to combine the crosses fingerprints.outType
- internalType
- public Output<T> outputValues()
Copyright © 2022. All rights reserved.