tensorflow.js安装与使用示例代码
时间: 2023-06-26 16:08:32 浏览: 230
安装 `tensorflow.js` 可以通过 `npm` 或者直接引入 `CDN` 文件进行安装。以下是使用 `npm` 的安装方法:
```
npm install @tensorflow/tfjs
```
安装完成后,在你的代码中引入 `tensorflow.js`:
```javascript
import * as tf from '@tensorflow/tfjs';
```
以下是一个简单的示例代码,展示如何使用 `tensorflow.js` 来进行张量计算:
```javascript
import * as tf from '@tensorflow/tfjs';
// 创建一个张量
const a = tf.tensor([1, 2, 3, 4]);
// 进行加法运算
const b = a.add(5);
// 打印结果
b.print();
```
这个例子中,我们首先创建了一个包含 `[1, 2, 3, 4]` 的张量 `a`,然后使用 `add()` 方法将其加上 `5`,得到了一个新的张量 `b`。最后,我们使用 `print()` 方法打印出了 `b` 的值,即 `[6, 7, 8, 9]`。
当然,`tensorflow.js` 不仅仅只能进行基础的张量计算,它还提供了许多高级的机器学习功能,例如模型训练、预测等等。如果你想深入学习 `tensorflow.js`,可以查看官方文档:https://www.tensorflow.org/js。
阅读全文