Function build_diag_union 

pub fn build_diag_union(tensors: &[&TensorDynLen]) -> AxisUnionFind

Build a union-find structure from a collection of tensors.

This helper is kept for callers that need to group diagonal axes by index ID. Numeric contraction currently keeps dense logical axes distinct and propagates diagonal result metadata separately.