使用張量流填充方法創建特定數據類型的張量

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

第一個是更好，因爲你在圖中使用更少的操作