给出embedding-layer函数的示例

好的，以下是一个简单的embedding-layer函数的示例：

import torch.nn as nn

class EmbeddingLayer(nn.Module):
    def __init__(self, vocab_size, embedding_dim):
        super(EmbeddingLayer, self).__init__()
        self.embedding = nn.Embedding(vocab_size, embedding_dim)
    
    def forward(self, inputs):
        embedded = self.embedding(inputs)
        return embedded

这个函数定义了一个embedding层，输入为一个vocab_size大小的词表和一个embedding_dim大小的嵌入维度。在前向传播中，输入是一个长为n的序列，每个元素都是词表中的索引，该函数输出一个(n, embedding_dim)的张量，其中每行都是对应输入序列中单词的嵌入。