【寒武纪硬件产品型号】必填*：
MLU270

【使用操作系统】必填*：
ubuntu

【使用驱动版本】必填*：
例如：v4.20.6

【出错信息】必填*：

pytorch模型中

t1 = torch.argmax(t, dim=-1)

出错

RuntimeError: Can not call cpu_data on an empty tensor

【当前已做了哪些信息确认】选填：
CPU能运行

【参考配置文档链接】选填

【相关日志文档】选填
如有，可附件

【出错代码链接】选填：
github的或gitee的代码的链接，