pseudoLoss<T> function

Tensor<Scalar> pseudoLoss<T>(
  1. Tensor<T> out
)

Implementation

Tensor<Scalar> pseudoLoss<T>(Tensor<T> out) {
  double sum = 0.0;
  for (int i = 0; i < out.data.length; i = i + 1) {
    sum = sum + out.data[i];
  }

  Tensor<Scalar> loss = Tensor<Scalar>(sum);
  loss.creator = Node(
      [out],
          () {
        for (int i = 0; i < out.data.length; i = i + 1) {
          out.grad[i] = out.grad[i] + loss.grad[0];
        }
      },
      opName: 'pseudo_sum_loss',
      cost: out.data.length
  );

  return loss;
}