Output ragged tensors from TF predict()
This is a very simple and dumb PR right now, but I think it's a lot less broken than our current implementation! Obviously don't merge it yet, I'm going to poke around to try to find edge cases and see if it resolves our memory leak issues.
_The documentation is not available anymore as the PR was closed or merged._<|||||>Closing this as we're going to fix this with an upstream PR to Keras instead