张量与 AutoGraph
张量与 AutoGraph
首先,我们导入 TensorFlow。由于名称有点长,我们经常使用短别名 tf 导入它。
import tensorflow as tf
张量表示数值的数组(可能是多维数组),在一个轴上,张量对应于(在数学上)矢量,在两个轴上,张量对应于一个矩阵;具有两个以上轴的张量没有特殊的数学名称。首先,我们可以使用 arange 创建一个行向量 x,该行向量 x 包含从 0 开始的前 12 个整数,尽管默认情况下它们创建为浮点数。张量中的每个值都称为张量的元素。例如,张量 x 中有 12 个元素。除非另有说明,否则新张量将存储在主存储器中,并指定用于基于 CPU 的计算。
x = tf.range(12)
x
<tf.Tensor: shape=(12,), dtype=int32, numpy=array([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11], dtype=int32)>
我们可以通过检查张量的 shape 属性来访问张量的形状(沿每个轴的长度)。
x.shape
TensorShape([12])
If we just want to know the total number of elements in a tensor, i.e., the product of all of the shape elements, we can inspect its size. Because we are dealing with a vector here, the single element of its shape is identical to its size.
tf.size(x)
<tf.Tensor: shape=(), dtype=int32, numpy=12>