python - 使用 tensorflow 填充方法创建特定数据类型的张量

Question

我正在尝试使用该tf.fill()方法创建不同数据类型（float16、float32、float64）的张量，类似于您可以使用numpy.full(). tf.constant() 会是一个合适的替代品吗？或者我应该将我的填充值创建为我希望它们成为的数据类型，然后将其插入内部的值持有者tf.fill()

Answer 1

您可以使用fill. 但是，结果类型将取决于value参数，您无法使用显式dtype参数控制结果类型。这确实与大多数其他 tensorflow 运算符有点不同。

tf.fill([2, 3], 9) # tensor with dtype=int23
tf.fill([2, 3], 9.0) # tensor with dtype=float32
# more explicit
tf.fill([2, 3], np.float64(9)) # tensor with dtype=float64

Answer 2

您可以提供您希望结果张量成为的数据类型的值，也可以在之后转换一个张量。

1. tf.fill((3, 3), 0.0)# 将是一个浮点数 32
2. tf.cast(tf.fill((3, 3)), tf.float32)# 也浮动 32

第一个更好，因为您在图中使用较少的操作