wasm-tensorflow-tfjs-core

如果你想在JavaScript中做这个呢？

import '@tensorflow/tfjs-backend-webgl'
import { setBackend } from '@tensorflow/tfjs-core'

await setBackend('wegbl')

只需添加此crate，你就可以在Rust中完成！

// Somehow import '@tensorflow/tfjs-backend-webgl'
use wasm_tensorflow_tfjs_core::{set_backend, BackendName};

let result = set_backend(BackendName::Webgl).await;