我正在将pytorch框架的代码转换为mindspore，目前正在处理Embedding层，torch代码如下：

input_data = self.embedding(input_data)

其中self为Controller的class，而我在转换为mindspore代码的时候，代码如下：

net = nn.Embedding() 

input_data = net(input_data)

则会报错：TypeError: __init__() missing 2 required positional arguments: 'vocab_size' and 'embedding_size'

也就是说，mindspore的embedding函数必须要给出'vocab_size' ，'embedding_size'的值，而我查阅torch的官方文档源代码不太肯定这个值给的是多少

想请教这里的mindspore代码应该怎么写，可以保证embedding层的转换正确