Why don’t you simply turn your tensorflow dataset to a list (since its a iterable, you should be able to do so in a one liner) and then solve problem from there. That is simply do :
tf_lst = list(tf_dataset)

now you have a list which you can simply incorporate into a new pytorch dataset and do as you wish!