tfdv.DecodeTFExample

View source on GitHub

Decodes serialized TF examples into Arrow RecordBatches.

examples A PCollection of strings representing serialized TF examples.
desired_batch_size Batch size. The output Arrow RecordBatches will have as many rows as the desired_batch_size.

A PCollection of Arrow RecordBatches.