1. 程式人生 > >【TensorFlow 2】矩陣基礎

【TensorFlow 2】矩陣基礎

隨機 orf 矩陣乘法 創建 flow run clas orm col

placeholder

placeholder為tf中的占位符,用來保存數據。語法為:

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

dtype:數據類型 shape:數據維度 name:名稱

返回類型:Tensor

使用方法:

data1 = tf.placeholder(tf.float32)
data2 = tf.placeholder(tf.float32)
dataAdd = tf.add(data1,data2)
with tf.Session() as sess:
    print(sess.run(dataAdd,feed_dict={data1:6,data2:2})) # feed_dict的作用是給使用placeholder創建出來的tensor賦值。

矩陣

創建矩陣並打印矩陣中的值:

data3 = tf.constant([[6,6]]) 
data4 = tf.constant([[2],[2]])
data5 = tf.constant([[3,3],[3,4],[5,6]])
with tf.Session() as sess:
    print(sess.run(data5)) #打印整體
    print(sess.run(data5[0])) #打印某一行
    print(sess.run(data5[:,1])) #打印某一列

矩陣乘法與普通乘法:

data3 = tf.constant([[6,6]])
data4 
= tf.constant([[3],[2]]) matMul = tf.matmul(data3,data4) matMul1 = tf.multiply(data3,data4) with tf.Session() as sess: print(sess.run(matMul)) # 矩陣乘法(1*2)*(2*1)->1*1 print(sess.run(matMul1)) # 普通乘法(1*2)*(2*1)->2*2

一些常用的矩陣生成:

mat0 = tf.zeros([2,3]) #全0矩陣
mat1 = tf.ones([3,1]) #
全1矩陣 mat2 = tf.fill([3,3],15) #填充矩陣 mat3 = tf.zeros_like(mat2) #mat3和mat2維度相同且全為0 mat4 = tf.linspace(0.0,2.0,10) # 把0.0~2.0 10等分存入矩陣 mat5 = tf.random_uniform([2,3],1,100) # 把1~100中的隨機數存入2*3的矩陣中

【TensorFlow 2】矩陣基礎