tensorflow安装可以直接通过命令行或者源码安装,在此介绍tensorflow8命令行安装如下:
安装tensorflow:sudo pip install --upgrade
https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.8.0-cp27-none-linux_x86_64.whl
另外,卸载tensorflow命令为: sudo pip uninstall tensorflow
tensorflow提供了函数使用说明文档,可以方便查找函数使用方法,地址为https://www.tensorflow.org/api_docs/python/array_ops/。
1.tensor类型的变量的索引:
例如我们定义矩阵:
import tensorflow as tf x=tf.zeros([2,3]) x1=tf.Variable(x) op= x1[0,1].assign(1)#对矩阵的0行1列赋值 with tf.Session() as sess: sess.run(tf.initialize_all_variables()) sess.run(op) print(sess.run(x1))
打印结果为:
[[ 0. 1. 0.]
[ 0. 0. 0.]]
2.矩阵相乘
对于二维矩阵相乘,函数为tf.matmul(M1,M2)
三维矩阵与三维矩阵相乘:
N=tf.batch_matmul(M1,M2)M1 = tf.Variable(tf.random_normal([2,3,4])) M2 = tf.Variable(tf.random_normal([2,4,5]))
得到N的大小为[2,3,5]
三维矩阵与二维矩阵相乘,即对三维矩阵的某一维度上的所有二维矩阵与同一个二维矩阵相乘:
M1 = tf.Variable(tf.random_normal([2,3,4])) M2 = tf.Variable(tf.random_normal([5,4])) N = tf.einsum('ijk,lk->ijl',M1,M2)]
得到N的大小为[2,3,5]