maeLossGPU function

GPUTensor<Scalar> maeLossGPU(
  1. GPUTensor<Vector> preds,
  2. GPUTensor<Vector> targets,
  3. CommandBuffer tape
)

Implementation

GPUTensor<Scalar> maeLossGPU(GPUTensor<Vector> preds, GPUTensor<Vector> targets, CommandBuffer tape) {
  GPUTensor<Vector> diff = subtractGPU<Vector>(preds, targets, tape);
  GPUTensor<Vector> absoluteDiff = absGPU<Vector>(diff, tape);
  GPUTensor<Scalar> totalError = sumGPU(absoluteDiff, tape);

  GPUTensor<Scalar> nScalar = GPUTensor<Scalar>(preds.shape[0].toDouble());

  return divideGPU<Scalar>(totalError, nScalar, tape);
}