jittor.nn.Embedding似乎仅支持单维度向量

我在使用jittor.nn.Embedding时发现,当我的输入加入了batch的N维度时,embedding就会报错,文档中的示例本机可以通过,但是当我将输入从[4]变为[1,4]时即出现了问题,想问一下是embedding存在问题吗?

您好,我刚刚尝试改变输入维度,没有出现您说的问题。

请您检查一下jittor是否是最新版(1.3.4.1),如果还有问题请给一段复现代码。

1.2.3.94,确实不是最新版本了