Add additional documentation to output_wrapper.py's transform_raw_features API.
Add additional documentation to output_wrapper.py's transform_raw_features API.

PiperOrigin-RevId: 371184362

tensorflow/transform