Pythonのtf.placeholder関数の使い方は何ですか。

2年 ago
陽, 向宇
1 minute

Pythonにおいて、tf.placeholder関数は、プレースホルダーとして定義されたtensorを作成し、TensorFlowの計算グラフ内で入力ノードとして使用するためのものです。計算グラフを実行する際には、具体的な値を渡すことで、このプレースホルダーを埋めることができます。

tf.placeholder関数の使い方は以下の通りです：

tf.placeholder(
    dtype,
    shape=None,
    name=None
)

パラメータ説明：

  1. dtypeは指定されたプレースホルダのデータ型を表し、tf.float32、tf.float64、tf.int32などが指定できます。
  2. 形状：指定プレースホルダーの形状を指定します。これはリストやタプルであることもありますし、Noneであることもあります。
  3. 名前：指定したプレースホルダーの名前、省略可能なパラメータ。

サンプルコード：

import tensorflow as tf

# 定义一个占位符
x = tf.placeholder(dtype=tf.float32, shape=(None, 2), name='input')

# 创建一个计算图
y = tf.reduce_sum(x, axis=1)

# 创建一个会话
with tf.Session() as sess:
    # 使用feed_dict参数传入具体的值来填充占位符
    result = sess.run(y, feed_dict={x: [[1, 2], [3, 4], [5, 6]]})
    print(result)  # 输出[3.  7. 11.]

上記のコードでは、最初にtf.placeholderを使用して、tf.float32のデータ型で形状が(None, 2)のプレースホルダーxが定義されました。これは、任意の行と2列の入力を受け入れることを表しています。次に、計算グラフでは、このプレースホルダーxを使用し、tf.reduce_sumを使用してxの2番目の次元を合計します。最後に、セッションで計算グラフを実行する際、feed_dictパラメーターを使用して具体的な値[[1, 2], [3, 4], [5, 6]]をプレースホルダーxに渡し、結果[3. 7. 11.]を計算します。

bannerAds