torch.nn.functional.mse_loss¶
- torch.nn.functional.mse_loss(input, target, size_average=None, reduce=None, reduction='mean', weight=None) Tensor[source][source]¶
测量元素级的均方误差，可选加权。
- 参数:
输入（Tensor）- 预测值。
目标（Tensor）- 真实值。
size_average（布尔值，可选）- 已弃用（使用 reduction）。
reduce（布尔值，可选）- 已弃用（使用 reduction）。
reduction（str，可选）- 指定应用于输出的降维方式：‘none’ | ‘mean’ | ‘sum’。‘mean’：取输出的平均值。‘sum’：输出将被求和。‘none’：不应用降维。默认：‘mean’。
weight（Tensor，可选）- 每个样本的权重。默认：None。
- 返回值:
均方误差损失（可选加权）。
- 返回类型:
