张量
tensorflow
中的基本数据是tensor
(张量),可以看作是多维数组或列表类型。
张量的创建
使用tf.constant()
创建张量,tf.constant()
的语法格式为:
tf.constant(value,dtype,shape)
value
用来指定数据,dtype
用来显式地声明数据类型,shape
用来指定数据的形状,
例如,要生成一个两行三列全为类型int32
的数字2的张量,可以使用以下代码:
import tensorflow as tf
a = tf.constant(3,dtype=tf.int32,shape=(2,3))
print(a)
但是由于tensor
中的整型数据默认是tf.int32
的,dtype
可以不用显式地指定。
tensorflow
所生成的张量都可以用numpy
方法转换成对应的数据,例:
import tensorflow as tf
a = tf.constant(3,dtype=tf.int32,shape=(2,3))
print(a)
print(a.numpy())
tensor
的数据类型有:
constant
函数的value
参数除了可以是数字外,还可以是numpy