I have a nn.Embedding layer. It maps integers to vectors of some dimension. We can think of it as a matrix times one-hot embedded inputs vectors i.e. torch.matmul(E, x). Given a vector y of the output dimension of E, I want to compute a probability distribution over the input space i.e. softmax(torch.matmul(E.T, y)). How do I do this?