tensorflow字面可以拆成tensor 和 flow两个词,实际应用中通过计算流图用于定义tensor之间的计算关系,然后通过run命令执行计算图的运算,得到需要的结果,本文主要介绍tensorflow的基础计算元素tensor。
tensor本质上是一个n维度向量或者矩阵,他们中的元素是具有基础数据类型(如int32, float32,string等)的数值,因此定义一个tensorflow需要两大要素:
- shape: 例如[n,k,m]表示一个n*k*m维的向量,维度也可以不确定,如[n,k,none],常用在某些tensor的维度事先不确定,运行时才能确定的场景中;dtype: 定义了tensor中元素的类型;
3 import tensorflow as tf 4 5 if __name__ == "__main__": 6 tensor_a = tf.placeholder(dtype = tf.float32, shape = [3,5,6]) 7 tensor_b = tf.placeholder(dtype = tf.float32, shape = [3,5,None]) 8 print("tensor_a: