你好,请问您这边用torch_mlu的版本是哪个?可以pip list | grep torch看一下
你好,请检查下你的输入shape是否有问题。
你好,经测试,Xv[:,i]这个切片放到cpu做可以解决这个问题,放到mlu上做会报上面的错误。您可以先这么规避进行后面工作,具体原因我们还在排查。
请问有进展吗
尊敬的开发者您好,我们会进行分析,有进展后回复您。
请登录后评论