利用标量值填充TensorFlow张量

tf.fill

fill(
    dims,
    value,
    name=None
)

参见指南:生成常量、序列和随机值>>参数值传感器

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

该操作创建一个形状 dims 的张量,并将其填充为值。

例如:

# Output tensor has shape [2, 3].
fill([2, 3], 9) ==> [[9, 9, 9]
                     [9, 9, 9]]

函数参数:

  • dims:int32 类型的张量。1-D:表示输出张量的形状。
  • value:张量。0-D(标量)。值来填充返回的张量
  • name:操作的名称(可选)。

函数返回值:

返回张量。与 value 具有相同的类型。

numpy 兼容性

相当于 np.full

发表评论

电子邮件地址不会被公开。 必填项已用*标注