Add more datapipe to functional API (#53123)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/53123
Test Plan: Imported from OSS
Reviewed By: VitalyFedyunin
Differential Revision: D26756638
Pulled By: ejguan
fbshipit-source-id: 6ff0eb6c7ee702056ff19eeb723949e4642f2784