Tensorflow tf.dynamic_partition矩阵拆分示例(Python3)