torch.nn.MSELoss
如果 reduce = False,那么 size_average 参数失效,直接返回向量形式的 loss
如果 reduce = True,那么 loss 返回的是标量。如果size_average为True,则除以参与计算loss的元素个数N,即取平均值;如果size_average为False,则为loss的总和,不除以N。
reduce和size_average的默认值为True
import torch
import numpy as np
# loss_fn = torch.nn.MSELoss
原创
2020-07-11 14:01:38 ·
722 阅读 ·
0 评论