《PyTorch深度学习实践》完结合集-哔哩哔哩】https://b23.tv/fSpROo
import math
for w in [0,1,2,3,4]:
loss_list = []
print('当权重w={}时,此时有:'.format(w))#注意format的使用方法
for x in [1,2,3]:
loss = math.pow((x*w - x*2),2)
loss_list.append(loss)#将计算的各个loss存入到列表中
print('x={}时,loss={}'.format(x,loss))
for i in [0,1,2]:
sum_loss = loss_list[i] + sum_loss
mean_loss = sum_loss/3
print('此时mean_loss={}'.format(mean_loss))
sum_loss=0
mean_loss=0