【PyTorch】RuntimeError: expected scalar type Float but found Long 已解决

主要问题:矩阵乘法时类型不一致。先看下面代码:

	A = torch.arange(20,dtype=torch.float32).reshape(5,4)
	a = torch.arange(4)
	A.shape,a.shape,torch.mv(A, a)

【PyTorch】RuntimeError: expected scalar type Float but found Long 已解决_第1张图片

出现该问题的主要原因是在A矩阵中已经设置了类型dtype=torch.float32,但在a向量中未设置,从下图可以看到,a的默认类型为int64,从而出现类型不一致的问题。

【PyTorch】RuntimeError: expected scalar type Float but found Long 已解决_第2张图片

	A = torch.arange(20,dtype=torch.float32).reshape(5,4)
	a = torch.arange(4,dtype=torch.float32)
	A.shape,a.shape,torch.mv(A, a)

请添加图片描述

你可能感兴趣的:(PyTorch,pytorch,深度学习,python,RuntimeError,矩阵乘法)