pytorch 中，使用自定义的损失函数训练模型

pytorch 中，使用自定义的损失函数训练模型

# 当Tensor算数操作（+， -，*， %等）中，有一个Tensor的resquire_grad=True,则该操作得到的Tensor
# 能够实现反向传播，自动求导的功能。
# 所以第一种自定义loss函数的方法就是使用tensor的math operation实现loss定义。

##############################################################
#1. 继承于nn.Module，nn.Module会维护一些其他变量和状态
# 在forward中实现loss定义，注意：
# 所有的数学操作使用tensor提供的 math operation
# 返回的tensor是0-dim的scalar
# 有可能会用到nn.functional中的一些操作

# 自定义MSEloss实现：
class My_loss(nn.Module):
	def __init__(self):
		super(My_loss