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