import tensorflow as tf

                                         [[0, 1, 2, 3], [1, 0, 2, 3], [2, 3, 1, 0], [2, 1, 0, 3], [0, 1, 2, 3]],
                                         num_partitions=4)  # after permute becomes 5,4,11, return all partitions 5,11
    node_a = tf.div(list_of_parts[0], list_of_parts[1])
    node_b = tf.divide(list_of_parts[2], list_of_parts[3])
