Solved by Kshitij_Jain in post #3 
              
              
                torch.einsum(“ik,ij->kj”, (adj, embed)) where adj is the given adjacency matrix of  size N * N and embed is the embedding matrix of size N * d.