Python tf.fill用法及代码示例

创建一个用标量值填充的张量。

用法

tf.fill(
    dims, value, name=None
)

另见tf.ones、tf.zeros、tf.one_hot、tf.eye。

此操作创建一个形状为 dims 的张量并用 value 填充它。

例如：

tf.fill([2, 3], 9)
<tf.Tensor:shape=(2, 3), dtype=int32, numpy=
array([[9, 9, 9],
       [9, 9, 9]], dtype=int32)>

tf.fill 在图形运行时评估并支持基于其他运行时的动态形状 tf.Tensors ，与 tf.constant(value, shape=dims) 不同，它将值嵌入为 Const 节点。

numpy 兼容性

类似于 np.full 。在 numpy 中，支持更多参数。将数字参数作为形状 (np.full(5, value)) 在 numpy 中用于指定一维形状结果是有效的，而 TensorFlow 不支持这种语法。