Tensorflow初心者です。
コードの書き方が全くわからなくて困っています。
もし良ければ教えて頂きたいです。

質問なんですけれど、
入力データは縦20横20の配列で、
1行に入る要素は'1'が1個、残りは'0'です。
それが20列あります。

答えは縦1横20の配列で、
これも要素は'1'が1個、'0'が19個です。

【例】5*5の場合

1　0　0　0　0
0　1　0　0　0
0　1　0　0　0　　→　　0　1　0　0　0
0　0　1　0　0
0　0　0　0　1
　入力データ　　　　　　答えデータ

　
入力データから、任意の'1'を選んで、
答えと一致させるプログラムを作りたいです。
例の場合だと、左から2番目を選んでほしいです。
（2行目か3行目かは問わない）

TensorflowのチュートリアルのMNISTをそのまま
入力データを上記のに替えて試しましたが
学習の正答率があがりませんでした。
私がやりたいのにはフィルターなどはいらないと
分かったものの、どうコードを書いて良いのか
全く分からない状態です。

教えてください、お願いします。