PyTorch遇到的error2

廖星宇 多项式拟合

UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number 

将pytorch更新到0.4.0最新版后对0.3.1版本代码会有如下警告，它在提醒用户下个版本这将成为一个错误

修改：

#    print_loss = loss.data[0]
    print_loss = loss.item()

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Fri Sep 14 16:19:59 2018

@author: lthpc
"""

import numpy as np
import matplotlib.pyplot as plt
import torch
from torch.autograd import Variable
from torch import nn,optim

def make_features(x):
    '''Builds features i.e. a matrix with columns [x, x^2, x^3].'''
    x = x.unsqueeze(1)
    return torch.cat([x ** i for i in range(1,4)],1)

W_target = torc