跳到主要内容

CPU 后端

2024年12月19日
柏拉文
越努力,越幸运

一、认识


CPU 后端TensorFlow.js 默认的后端,当其他后端不可用时,TensorFlow.js 会自动使用 CPU 后端。它通过 JavaScript 引擎在 CPU 上执行所有计算。

CPU 后端 无需额外硬件支持,适用于低端设备或没有硬件加速如 GPU 或者 WebAssembly支持的环境, 来进行简单的机器学习任务或者低计算量任务。相对于 WebGLWebAssembly 后端,CPU 后端的计算速度较慢,尤其是当处理大规模的数据时,可能会成为性能瓶颈。

二、原理


CPU 后端直接在 CPU 上执行计算,不依赖于 GPUWebAssembly。它通过 JavaScript 本身的计算能力来完成张量操作。