Add task id to multiproblem features.
Add task id to multiproblem features.

PiperOrigin-RevId: 212535366

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor