如何解决与 GPU 并行构建 Tensorflow 数据
假设我想构建一个巨大的张量。在像 CUDA 这样的东西中，或者任何涉及在 GPU 中构建的东西，我们可以有这样的东西：
gpu.createKernel(function(a: number[][],b: number[][]) {
return complicatedLogic ? 1 : 0;
}).setoutput([512,512]);
结果中的每个“元素”将在 GPU 上并行运行。如何在 tensorflow.js 中实现类似的功能？
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
