square_error_cost

paddle.nn.functional.square_error_cost ( input, label ) [源代码]

该OP用于计算预测值和目标值的方差估计。

对于预测值input和目标值label,公式为:

square_error_cost - 图1

参数:

  • input (Tensor) - 预测值,维度为

    square_error_cost - 图2

    的多维Tensor,其中最后一维D是类别数目。数据类型为float32或float64。

  • label (Tensor) - 目标值,维度为 [N1,N2,…,Nk,D][N1,N2,…,Nk,D] 的多维Tensor,其中最后一维D是类别数目。数据类型为float32或float64。

返回:Tensor, 预测值和目标值的方差

代码示例

  1. import paddle
  2. input = paddle.to_tensor([1.1, 1.9])
  3. label = paddle.to_tensor([1.0, 2.0])
  4. output = paddle.nn.functional.square_error_cost(input, label)
  5. print(output)
  6. # [0.01, 0.01]