pub fn index_for_dimension_mut( indices: &mut [TensorDimensionIndexSelection], dimension: u32, ) -> Option<&mut u64>