No, I don’t want to implement CBOW. I just want to know the corresponding architecture matching  with nn.Embedding and if I understand it right or not. So, it looks like the nn.Embedding is equal to the red square part in the picture.