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 不支持這種語法。