Is there something like torch.utils.data.TensorDataset(*tensors) for TensorFlow/Keras?

Question

I´m "translating" a notebook made in Pytorch to one made in Keras. And they use that app to pack the data from a tensor into the dataset that will be used for the