TensorFlowTensorFlow.js智能前端实战:Tensorflow.js简介
// 引入TensorFlow.js库
import * as tf from '@tensorflow/tfjs';
// 创建一个Tensor,这里是一个1x3的浮点数矩阵
const tensor = tf.tensor2d([[1.0, 2.0, 3.0]]);
// 打印tensor的形状和数据
tensor.print();
// 执行tensor的加法操作
const result = tensor.add(tf.tensor2d([[4.0, 5.0, 6.0]]));
// 打印加法结果
result.print();
// 释放tensor和result占用的内存
await tensor.dispose();
await result.dispose();
这段代码演示了如何在TensorFlow.js中创建一个Tensor,执行基本的操作,并在最后释放所占用的资源。代码简洁,注重于教学核心功能,对于学习者来说具有很好的教育价值。
评论已关闭