Read dataset from TFRecord format

FYI there is also a tfrecord reader in xla:


Have you compared this implementation to other ones? I guess it is good enough, so I will give it a try. :ok_hand:

I’ve tried a few then but could not get anything working reasonably with multiple files, unfortunately…
I wonder if we can actually use and convert like torch.from_numpy(tf_tensor.numpy())

Please let us know if you find a good way.
I think it’s a common request, and I (and perhaps many more) would very interested to know!

1 Like

Cool,thank you,guys。 :grin: :grin: