fn dimension_name(shape: &[TensorDimension], dim_idx: u32) -> String