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

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

【使用驱动版本】必填*：v4.9.8

【出错信息】必填*：

我想要在MLU270上进行张量修改时出错，例如：

state = torch.zeros([1, 200]).to(ct.mlu_device())

state[0] = 1  # 这里报错

我只有这样才能成功运行：

state[0] = torch.FloatTensor(1).to(ct.mlu_device())

那是不是所有的张量修改都必须进行这样的步骤？

另外，诸如这样的张量修改有没有办法支持（上述trick不可行）：probs[state[0] == 1]  = 1，其中probs已经在MLU上

如果上面一条可以解决，既然MLU270不支持torch.argmax()算子的话，我们开发者能不能自己写一个？