Fix bug that prevents the use of a single layer.
Fix bug that prevents the use of a single layer.

If you build a functional model with a list of one output, the result is not returned as a list. This change wraps the result in a list so that the rest of the code works either way.

PiperOrigin-RevId: 290896517

tensorflow/docs,tensorflow/docs,tensorflow/docs