softmax static method

List<double> softmax(
  1. List<double> logits
)

Applies softmax to raw logits.

Implementation

static List<double> softmax(List<double> logits) {
  final maxLogit = logits.reduce((a, b) => a > b ? a : b);
  final exps = logits.map((l) => _exp(l - maxLogit)).toList();
  final sum = exps.reduce((a, b) => a + b);
  return exps.map((e) => e / sum).toList();
}