我们可以通过传入列表或 NumPy 数组来新建变量和常量类型的张量
(1)查看 TensorFlow 版本:
(2)定义形状为 (3, 3) 的二维变量:
(3)定义现状为 (2,2) 二维常量
仔细观察,你会发现输出包含了张量的 3 部分属性,分别是
-
形状
shape
-
数据类型
dtype
, -
对应的 NumPy 数组。
你还可以直接通过 .numpy()
输出张量的 NumPy 数组。
上面我们已经介绍了常量张量,这里再列举几个经常会用到的新建特殊常量张量的方法:
-
tf.zeros
:新建指定形状且全为 0 的常量 Tensor -
tf.zeros_like
:参考某种形状,新建全为 0 的常量 Tensor -
tf.ones
:新建指定形状且全为 1 的常量 Tensor -
tf.ones_l