tf.placeholder(
dtype,
shape=None,
name=None
)
- dtype:数据类型。常用的是tf.float32,tf.float64等数值类型
- shape:数据形状。默认是None,就是一维值,也可以是多维(比如[2,3], [None, 3]表示列是3,行不定)
- name:名称
例2 计算矩阵相乘
import tensorflow as tf
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
a1=tf.placeholder(tf.float32)
a2=tf.placeholder(tf.float32)
z=tf.matmul(a1,a2)
import numpy as np
with tf.compat.v1.Session() as sess:
rand_array = np.random.rand(2,2)
print(sess.run(z, feed_dict={a1: rand_array,a2:[[1,1],[2,3]]})) # Will succeed.