[Task]: Create a helper function fill_in_missing for TFT Criteo tests

[AnandInguva]

What needs to happen?

This function should accept a rank 2 SparseTensor, and a default value, and return a rank 1 Tensor. It will assume the input is from a VarLenFeature and has dimensions [batch_size, 0] or [batch_size, 1] depending on the max size of the feature over the batch. It's assumed each feature has 0 or 1 values (0 for missing, 1 for present).

It will emit a Tensor which is constructed using the code

  feature = tf.sparse_to_dense(
      feature.indices, [feature.dense_shape[0], 1], feature.values,
      default_value=-1)
  feature = tf.squeeze(feature, axis=1)

Issue Priority

Priority: 3 (nice-to-have improvement)

Issue Components

• Component: Python SDK
• Component: Java SDK
• Component: Go SDK
• Component: Typescript SDK
• Component: IO connector
• Component: Beam examples
• Component: Beam playground
• Component: Beam katas
• Component: Website
• Component: Spark Runner
• Component: Flink Runner
• Component: Samza Runner
• Component: Twister2 Runner
• Component: Hazelcast Jet Runner
• Component: Google Cloud Dataflow Runner

[smeet07]

.take-issue

[smeet07]

@AnandInguva where should I put this helper function?

[AnandInguva]

This is the place that needs this feature.

beam/sdks/python/apache_beam/testing/benchmarks/cloudml/criteo_tft/criteo.py

Line 135 in 1fc8e5c

# TODO(https://github.com/apache/beam/issues/24902):

[smeet07]

R: @AnandInguva