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