Tensorflow.js tf.setBackend()函数

Tensorflow.js是谷歌开发的一个开源库，用于在浏览器或节点环境中运行机器学习模型和深度学习神经网络。它还可以帮助开发人员用 JavaScript 语言开发 ML 模型，并且可以直接在浏览器或 Node.js 中使用 ML。

tf.setBackend()函数 用于设置当前后端名称。

句法：

tf.setBackend( backendName )

参数：

• backendName：后端的名称。

返回值：返回 Promise

示例 1：在此示例中，我们将后端设置为字符串“cpu”。然后将其设置为字符串“webgl”并删除后端返回“null”字符串。

Javascript

// 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”。

Javascript

// 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