Make NestedMapFromBatchedOutputs public, add more documentation.
Make NestedMapFromBatchedOutputs public, add more documentation.

Make shape assertion check aggregate all errors so that the error
message contains all violations.

Fix shape specification when batched_input is True.

PiperOrigin-RevId: 350631628

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo