pub fn balance_partitions(
tensor: &Tensor,
num_partitions: usize,
initial_partitioning: &[usize],
communication_scheme: CommunicationScheme,
memory_limit: Option<f64>,
) -> (Vec<usize>, f64)Expand description
Balances partitions using a genetic algorithm. Finds the partitioning that reduces the total contraction cost.