RuntimeError: expected scalar type Long but found Float

 x1 = t.arange(0, 20).view(-1, 1)
 y1 = x1.mm(w) + b.expand_as(x1)

报错RuntimeError: expected scalar type Long but found Float

debug发现是x1.mm(w)的问题,尝试通过更改w的类型解决(w.long()),虽然不报错,但是数据要求float类型,更改后会影响实验结果;尝试更改x1的类型,查询后发现torch.arange输出的是long型,torch.range输出float型,但是后者已经弃用,通过更改torch.arange参数dtype=torch.float解决

x1 = t.arange(0, 20, dtype=t.float).view(-1, 1)

你可能感兴趣的:(torch,pytorch,深度学习,人工智能)