Expected isFloatingType(grads[i].type().scalarType()) to be true, but got false._runtimeerror: expected isfloatingtype(grads[i].typ

作者：小蓝xlanll | 2024-05-31 02:19:13

踩

runtimeerror: expected isfloatingtype(grads[i].type().scalartype()) to be tr

错误原因

在进行张量梯度求解时，传入数据类型不对

解决方法

指定传入张量类型为 float 类型即可

错误代码示例


import torch
 
# 第一步：创建 tensor
x = torch.ones(2,2,requires_grad=True)
print(x)
 
# 第二步：对 tensor 做处理
y = x**2
print(y)
 
# 第三步：求梯度
y.backward(torch.tensor([[3,2],[2,2]]))
print(x.grad)

正确代码示例


import torch
 
# 第一步：创建 tensor
x = torch.ones(2,2,requires_grad=True)
print(x)
 
# 第二步：对 tensor 做处理
y = x**2
print(y)
 
# 第三步：求梯度
y.backward(torch.tensor([[3,2],[2,2]],dtype=float)) # 这里加了一个类型
print(x.grad)

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小蓝xlanll/article/detail/649862