Tensorflow.js tf.setBackend()用法及代碼示例

Tensorflow.js是由Google開發的開源庫，用於在瀏覽器或節點環境中運行機器學習模型和深度學習神經網絡。它還可以幫助開發人員使用JavaScript語言開發ML模型，並且可以直接在瀏覽器或Node.js中使用ML。

tf.setBackend()函數用於設置當前的後端名稱。

用法:

tf.setBackend( backendName )

參數：

• backendName:後端名稱。

返回值：它返回Promise <boolean>

範例1：在此示例中，我們將後端設置為字符串“cpu”。然後將其設置為字符串“webgl”，並刪除後端，返回“null”字符串。

// Setting the backend
tf.setBackend("cpu")
console.log(tf.getBackend())
  
tf.setBackend("webgl")
tf.removeBackend("webgl")
console.log(tf.getBackend())

輸出：

cpu
null

範例2：在此示例中，我們將後端設置為字符串“cpu”。刪除了後端，並在控製台中返回“null”。我們將後端設置為“webgl”，並將後端作為字符串“webgl”返回。

// Setting the backend
tf.setBackend("cpu")
tf.removeBackend('cpu')
console.log(tf.getBackend())
  
tf.setBackend("webgl")
  
// Getting the backend using
// getBackend method
console.log(tf.getBackend())

輸出：

null
webgl

參考： https://js.tensorflow.org/api/latest/#setBackend