【Python】TensorFlow入門

 

こんにちは。のっくん(@yamagablog)です。

 

今日の記事では、TensorFlowというライブラリを使って数値を計算するプログラムを書いてみます。

 

TensorFlowはディープラーニング用のライブラリですが、その前に“簡単な演算”をしてどういった仕組みで動いているのか確認したいと思います。

 

「TensorFlowって何だか難しそう」

 

「TensorFlowのコードを読んで挫折したから基本をもう一回勉強したい

 

そう思っている方に読んでもらえればと思います。

 

スポンサーリンク

 

準備

 

まずはライブラリをインストールします。

 

 

 ハローワールド

 

手始めにハローワールドをしてみましょう。

 

 

Pythonだと1行でハローワールドできるのに、tensorflowだと5行も書かないといけないんです。(汗)

 

constantと言うのは定数です。セッションを開始して、runメソッドを呼び出すことでHelloWorldを出力しています。

 

足し算と掛け算

 

簡単な演算をしてみます。

 

 

実行すると500と出力されます。

 

演算を定義している部分を「データフローグラフ」と言います。このグラフを実行するためにセッションを開始しています。

 

プレースホルダを使う

 

TensorFlowでよく使われるのが“プレースホルダ”と言われるものです。

 

これはデータフローグラフの構築時には実際に値を入れず入れ物だけを用意して置いて、セッションを実行するときに実際の値を入れる仕組みです。

 

入れ物(プレースホルダ)を5個用意してみます。

 

 

[30 60 90 120 150] が出力されます。

 

任意のサイズを指定

 

要素数を5で固定しましたが、Noneを使うと任意のサイズを指定できます。

 

 

実行すると、listの中身が3倍された結果が得られます。

 

 

まとめ

 

今日のまとめ
・TensorFlowではデータフローグラフを作ってセッションを走らせて実行する
・プレースホルダーを使うと実行時に値を割り当ててくれる
・プレースホルダーでは任意のサイズを指定できる

 

参考

 

ABOUTこの記事をかいた人

のっくん

理系院卒で大企業の研究所に就職。 趣味はプログラミング、レアジョブ英会話、筋トレ、旅行。 Twitter:@yamagablog