tensorflow/docs/api/tfjs

TensorFlow.js 是一个开源库，它允许你使用 JavaScript 在浏览器或 Node.js 中运行 TensorFlow 模型。以下是一些关于 TensorFlow.js 的基本信息和资源。

快速开始

要开始使用 TensorFlow.js，首先需要确保你的开发环境已经安装了 Node.js 或浏览器。

• TensorFlow.js 安装指南

API 文档

这里是 TensorFlow.js 的 API 文档，包含了所有可用的函数和类。

• TensorFlow.js API 文档

示例

以下是一个简单的 TensorFlow.js 示例，演示了如何加载和运行一个模型：

import * as tf from '@tensorflow/tfjs';

async function loadModel() {
  const model = await tf.loadLayersModel('https://storage.googleapis.com/tfjs-models/tfjs/mnist/model.json');
  const tensor = tf.tensor2d([1, 2, 3, 4, 5, 6, 7, 8, 9, 0], [1, 28, 28, 1]);
  const prediction = model.predict(tensor);
  console.log(prediction);
}

loadModel();

学习资源

• TensorFlow.js 教程

社区

TensorFlow.js 有一个活跃的社区，你可以在以下地方找到帮助和资源：

• TensorFlow.js GitHub 仓库
• TensorFlow.js Stack Overflow 标签

希望这些信息能帮助你更好地了解和使用 TensorFlow.js！