It does not matter if feature vectors are created using TF, at the end they are all float values. The difference only comes to importing them to your code.
I don’t have the code but the general approach is

Create a vocab for your data
Create a dict or something for the embeddings
Use your favourite model, while using nn.Embedding as the first layer in your model