Generates sparse cross from a list of sparse and dense tensors.
tf.raw_ops.SparseCross( indices, values, shapes, dense_inputs, hashed_output, num_buckets, hash_key, out_type, internal_type, name=None )
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")))
Args | |
---|---|
indices | A list of Tensor objects with type int64 . 2-D. Indices of each input SparseTensor . |
values | A list of Tensor objects with types from: int64 , string . 1-D. values of each SparseTensor . |
shapes | A list with the same length as indices of Tensor objects with type int64 . 1-D. Shapes of each SparseTensor . |
dense_inputs | A list of Tensor objects with types from: int64 , string . 2-D. Columns represented by dense Tensor . |
hashed_output | A bool . If true, returns the hash of the cross instead of the string. This will allow us avoiding string manipulations. |
num_buckets | An int that is >= 0 . It is used if hashed_output is true. output = hashed_value%num_buckets if num_buckets > 0 else hashed_value. |
hash_key | An int . Specify the hash_key that will be used by the FingerprintCat64 function to combine the crosses fingerprints. |
out_type | A tf.DType from: tf.int64, tf.string . |
internal_type | A tf.DType from: tf.int64, tf.string . |
name | A name for the operation (optional). |
Returns | |
---|---|
A tuple of Tensor objects (output_indices, output_values, output_shape). | |
output_indices | A Tensor of type int64 . |
output_values | A Tensor of type out_type . |
output_shape | A Tensor of type int64 . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/SparseCross