1.6　TensorFlow JS的输入输出

本节用输入输出模型来分析TensorFlow JS的工作原理。

TensorFlow JS是开源机器学习软件库TensorFlow的JavaScript实现，擅长各种感知和语言理解等任务。它支持用GPU的片元着色器来实现硬件加速。

用户以Tensor的形式输入数据。输入的数据被当作GPU的纹理传递给GPU。GPU对纹理数据进行运算后，将结果写入输出的纹理。由于输出的纹理是存在GPU里面的，CPU需要用读回操作（gl. readPixels）将数据输出到输出Tensor里面。和本书要讨论的其他GPU处理模型不同的是，TensorFlow JS的目标是为了计算，所以TensorFlow JS没有将结果输出到一个具体的窗口。其输入输出模型如图1-6所示。

图1-6　TensorFlow的输入输出模型

测试的例子，如程序清单1-11所示。

程序清单1-11　TensorFlow JS实现的多个数求乘积