pytorch常用损失函数
损失函数的基本用法: criterion = LossCriterion() #构造函数有自己的参数 loss = criterion(x, y) #调用标准时也有参数 得到的loss结果已经对mini-batch数量取了平均值 1.BCELoss(二分类) CLASS torch.nn.BCELoss(weight=None, size_average=None, reduce=None, reduction='mean') 创建一个衡量目标和输出之间二进制交叉熵的criterion unreduced loss函数(即reduction参数设置为'none')为: N表示batch size,x n 为输出,y n 为目标 如果reduction不为'none'(默认设为'mean'),则: 即默认情况下,loss会基于 element求 平均值,如果 size_average=False 的话, loss 会被累加。 这是用来测量误差error的重建,例如一个自动编码器 。注意 0<=target[i]<=1。 参数: weight ( Tensor , 可选 ) – 每批元素损失的手工重标权重。如果给定,则必须是一个大小为“nbatch”的张量。 size_average ( bool , 可选 ) – 弃用(见 reduction 参数)。默认情况下,设置为True