TensorFlowとPythonで行列の全要素・特定の行を合計する方法

TensorFlowは、Googleが提供する機械学習向けのオープンソースフレームワークです。Pythonと組み合わせて使用することで、各種アルゴリズムやディープラーニングアプリケーションの実装が可能になり、研究用途から本番環境まで幅広く活用されています。

TensorFlowには高度な最適化技術が組み込まれており、複雑な数値計算を高速に処理できます。これは内部でNumPyと多次元配列を使用しているためです。この多次元配列は「テンソル（Tensor）」と呼ばれます。深層ニューラルネットワークの構築を強力にサポートし、高いスケーラビリティを備えている点も特徴です。さらに、一般的なデータセットが多数同梱されており、GPUによる演算処理やリソース管理の自動化にも対応しています。

TensorFlowのインストール

Windows環境では、以下のコマンドで「tensorflow」パッケージをインストールできます。

pip install tensorflow

テンソルとは

テンソルはTensorFlowにおける基本的なデータ構造であり、データフローグラフ（Data Flow Graph）内のエッジを接続する役割を担います。実体は多次元配列、あるいは単純なリストにすぎません。

本記事のコードはJupyter Notebookで動作確認しています。Jupyter Notebookへのインストールも同様に「pip install tensorflow」で行えます。

行列の合計を求めるサンプルコード

import tensorflow as tf
import numpy as np

matrix_1 = tf.Variable([[1,2,3],[4,5,8],[9,10,0]])
print("The matrix is ")
print (matrix_1)

print("The sum of all elements ")
result = tf.reduce_sum(matrix_1)
print(result)
print("The sum of specific rows is")
result = tf.reduce_sum(matrix_1, 1)
print(result)

実行結果

The matrix is
<tf.Variable 'Variable:0' shape=(3, 3) dtype=int32, numpy=
array([[ 1, 2, 3],
   [ 4, 5, 8],
   [ 9, 10, 0]], dtype=int32)>
The sum of all elements
tf.Tensor(42, shape=(), dtype=int32)
The sum of specific rows is
tf.Tensor([ 6 17 19], shape=(3,), dtype=int32)

コードの解説

• 必要なパッケージをインポートし、扱いやすいように別名（エイリアス）を付与します。
• NumPyパッケージを使って行列を作成します。
• 「reduce_sum」関数を使用すると、行列内のすべての値の合計を求められます。
• 「reduce_sum」の第2引数に軸（axis）を指定すると、その軸に沿った各行ごとの合計が計算されます。上記の例ではaxis=1を指定しているため、行ごとの合計 [6, 17, 19] が得られます。
• 計算結果はコンソールに出力して確認します。

このように、tf.reduce_sumを使いこなすことで、全体の合計だけでなく行・列単位の集計も柔軟に行えます。列ごとの合計を求めたい場合は、第2引数に0を指定してください。