Tensorflow 是否能够在 JavaScript 中操作 Uint8Array

是否能够在 JavaScript 中操作 Uint8Array"/>

Tensorflow 是否能够在 JavaScript 中操作 Uint8Array

我是 Tensorflow 的新手，如果这很明显，我很抱歉，但这是我的问题。

假设我在 JavaScript 中有两个 Uint8Arrays 和一个我想要匹配的子数组：

const a1 = new Uint8Array([1, 2, 3, 3])
const a2 = new Uint8Array([2, 0, 4, 4])
const matcher = new Uint8Array([4, 4])

在 NodeJS 中，我有一个函数可以循环遍历两个数组，如果该块与“匹配器”块匹配，我可以从一个到另一个交换：

// a2[2] and a2[3] match, so swap those from a2
a1[i] = a2[i]

这很有效并且很容易编写，对于小集合来说速度非常快，但是对于非常大的数组来说这很慢（想想图像像素）。我认为如果我可以使用 GPU 进行此计算可能会更快。我想知道具有GPU加速功能的Tensorflow是否可以做这样一个自定义数组组合器？

我做了很多谷歌搜索，但到目前为止还没有找到任何东西。