欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

TypeError: rsub() received an invalid combination of arguments - got (Tensor, numpy.ndarray), but

程序员文章站 2022-05-27 12:16:47
...

TypeError: rsub() received an invalid combination of arguments - got (Tensor, numpy.ndarray), but expected one of:

  • (Tensor input, Tensor other, *, Number alpha)
  • (Tensor input, Number other, Number alpha)

在写Logistic回归最后,将模型的参数取出来,并将直线画出来,报错

原因:张量变量不能在一个式子中同用

原:

w0, w1 = logistic_model.lr.weight[0]
w0 = w0.item()
w1 = w1.item()
b = logistic_model.lr.bias.data[0]
plot_x = np.arange(30, 100, 0.1)
plot_y = (-w0 * plot_x - b) / w1
plt.plot(plot_x, plot_y)

修正:

w0, w1 = logistic_model.lr.weight[0]
w0 = w0.item()
w1 = w1.item()
b = logistic_model.lr.bias.data[0]
plot_x = np.arange(30, 100, 0.1)
plot_y = (-w0 * torch.from_numpy(plot_x) - b) / w1  #
plt.plot(plot_x, plot_y.numpy())  #