Skip to content
Discussion options

You must be logged in to vote

I am not totally sure I understand the issue, but if your assingment between x and p changes dynamically, you somehow need to store this assignment somewhere and use it when accessing embeddings, e.g.:

out = self.embedding(x[assign_index[0]]) + self.embedding(p[assign_index[1]])

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by Sanjaykrishnamurthy
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants