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

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

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