矩阵乘法
-
【PyTorch】RuntimeError: expected scalar type Float but found Long 已解决
主要问题:矩阵乘法时类型不一致。先看下面代码: A = torch.arange(20,dtype=torch.float32).reshape(5,4) a = torch.ar…
主要问题:矩阵乘法时类型不一致。先看下面代码: A = torch.arange(20,dtype=torch.float32).reshape(5,4) a = torch.ar…