【TensorFlow】Tensor情報の確認方法

Googleによって開発されている機械学習ライブラリであるTensorFlowで、Tensor（テンソル）の情報を確認する方法を解説します。

Tensorの情報の確認方法

TensorFlowで使用するTensorには型等の各種情報があり、簡単に参照することができます。Tensorの情報について確認する方法を紹介します。

Tensorの型を確認する dtype

Tensorの型を確認する場合には、以下のようにdtypeプロパティを確認します。

import tensorflow as tf

tensor = tf.constant([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Tensorの型を確認する dtype
print(tensor, "\n")
print(tensor.dtype)

【実行結果】
tf.Tensor(
[[1 2 3]
 [4 5 6]
 [7 8 9]], shape=(3, 3), dtype=int32) 

<dtype: 'int32'>

Tensorの要素数を確認する tf.size

Tensorの要素数を確認する場合、sizeというプロパティはないため以下のようにtf.size関数を使用して確認します。

import tensorflow as tf

tensor = tf.constant([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Tensorの要素数を確認する tf.size
print(tensor, "\n")
print(tf.size(tensor))

【実行結果】
tf.Tensor(
[[1 2 3]
 [4 5 6]
 [7 8 9]], shape=(3, 3), dtype=int32) 

tf.Tensor(9, shape=(), dtype=int32)

Tensorの形状を確認する shape

Tensorの形状を確認する場合には、以下のようにshapeプロパティを確認します。

import tensorflow as tf

tensor = tf.constant([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Tensorの形状を確認する shape
print(tensor, "\n")
print(tensor.shape)

【実行結果】
tf.Tensor(
[[1 2 3]
 [4 5 6]
 [7 8 9]], shape=(3, 3), dtype=int32) 

(3, 3)

Tensorの階数を確認する ndim

Tensorの階数を確認する場合には、以下のようにndimプロパティを確認します。

import tensorflow as tf

tensor = tf.constant([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Tensorの次元数を確認する ndim
print(tensor, "\n")
print(tensor.ndim)

【実行結果】
tf.Tensor(
[[1 2 3]
 [4 5 6]
 [7 8 9]], shape=(3, 3), dtype=int32) 

2

まとめ

Googleによって開発されている機械学習ライブラリであるTensorFlowで、Tensor（テンソル）の情報を確認する方法を紹介しました。

• dtype：Tensorの型を確認する
• tf.size：Tensorの要素数を確認する（プロパティは持っていないため関数で確認）
• shape：Tensorの形状を確認する
• ndim：Tensorの階数を確認する

TensorFlowを用いたプログラミングでは、演算におけるTensorがどういったものかをよく考えてプログラミングする必要があります。Tensor情報は随時確認しながらプログラミングをする必要があるため、使い方をしっかり覚えておきましょう。

【Python Tech】プログラミングガイド本サイトで紹介しているPython入門、応用に関する記事のまとめページです。Python文法の入門から高度な内容、おすすめの書籍についても紹介しています。また、機械学習等のデータ分析やGUI開発、近年話題の生成AI等のトピックスについても含んでいます。...

Pythonによるディープラーニング

posted with ヨメレバ

François Chollet/巣籠悠輔/株式会社クイープ マイナビ出版 2022年03月23日頃

楽天ブックス

Amazon

ホッシー

システムエンジニア

Contact

はじめまして。当サイトをご覧いただきありがとうございます。 私は製造業のメーカーで、DX推進や業務システムの設計・開発・導入を担当しているシステムエンジニアです。これまでに転職も経験しており、以前は大手電機メーカーでシステム開発に携わっていました。

プログラミング言語はこれまでC、C++、JAVA等を扱ってきましたが、最近では特に機械学習等の分析でも注目されているPythonについてとても興味をもって取り組んでいます。これまでの経験をもとに、Pythonに興味を持つ方のお役に立てるような情報を発信していきたいと思います。どうぞよろしくお願いいたします。

※キャラクターデザイン：ゼイルン様