Function partition_tensor_network 

pub fn partition_tensor_network(tn: Tensor, partitioning: &[usize]) -> Tensor

Partitions the tensor network based on the partitioning vector that assigns each vector to a partition.