verifyLayerNorm function

void verifyLayerNorm()

Implementation

void verifyLayerNorm() {
  final x = Tensor.fromList([1, 4], [10.0, 2.0, 5.0, 3.0]);

  final y = x.normalize(eps: 1e-6);
  final loss = y.sum();
  loss.backward();

  final grads = x.grad;
  bool ok = grads.every((g) => closeEnough(g, 0.0, 1e-5));
  print("LAYERNORM: ${ok ? '✅ PASS' : '❌ FAIL'}");
}