将C ++ TensorFlow与WebAssembly后端一起使用

问题描述 投票:0回答:2

出于实验目的,是否可以将TensorFlow的C++ versionWebAssembly backend中的tfjs一起使用?

如果可能,那么该怎么做?

tensorflow webassembly tensorflow.js
2个回答
0
投票

您链接到的WebAssembly后端是TensorFlow的C ++版本,可使用Emscripten编译为WebAssembly。它在GitHub上有用法说明。


0
投票

您需要下载tfjs的wasm软件包。由于后端是异步的,因此您将等到后端准备好后再调用包含张量操作的main函数。请注意,wasm后端尚不支持所有操作]

// Import @tensorflow/tfjs or @tensorflow/tfjs-core
import * as tf from '@tensorflow/tfjs';
// Adds the WASM backend to the global backend registry.
import '@tensorflow/tfjs-backend-wasm';
// Set the backend to WASM and wait for the module to be ready.
tf.setBackend('wasm').then(() => main());
© www.soinside.com 2019 - 2024. All rights reserved.