tensorflow 实现矩阵逻辑变换（tf.gather/tf.gather_nd）