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

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())  #