用tf.one_hot函数写任一坐标为1的二维张量