CPU 后端
2024年12月19日
一、认识
CPU
后端 是 TensorFlow.js
默认的后端,当其他后端不可用时,TensorFlow.js
会自动使用 CPU
后端。它通过 JavaScript
引擎在 CPU
上执行所有计算。
CPU
后端 无需额外硬件支持,适用于低端设备或没有硬件加速如 GPU
或者 WebAssembly
支持的环境, 来进行简单的机器学习任务或者低计算量任务。相对于 WebGL
和 WebAssembly
后端,CPU
后端的计算速度较慢,尤其是当处理大规模的数据时,可能会成为性能瓶颈。
二、原理
CPU
后端直接在 CPU
上执行计算,不依赖于 GPU
或 WebAssembly
。它通过 JavaScript
本身的计算能力来完成张量操作。