我在将pytorch的代码转换为mindspore，其中在embedding层遇见代码：

input_data = self.embedding(input_data)

其中input_data为int64类型，torch的embedding能够处理

我根据ms的文档改为了：

net = nn.Embedding(vocab_size=self.hidden_size, embedding_size=self.embedding_size,dtype=ms.int64) 

input_data = net(input_data)

在使用AutoDL进行debug的时候，发现定义出来的net的dtype仍为float32，导致了类型错误

猜测是Mindspore的embedding层内部没有根据输入的dtype=int64这个参数进行更改，请问该怎么解决呀？