Function assert_close

cake-core/tests/unit_tests/test_backend_ops.rs:10–20

(a: &Tensor, b: &Tensor, tol: f32)

}

fn assert_close(a: &Tensor, b: &Tensor, tol: f32) {
    let diff = (a - b).unwrap().abs().unwrap();
    let max_diff: f32 = diff
        .flatten_all()
        .unwrap()
        .max(0)
        .unwrap()
        .to_scalar()
        .unwrap();
    assert!(max_diff < tol, "max diff {max_diff} exceeds tolerance {tol}");
}

// ── linear_forward ─────────────────────────────────────────────────────