Tensorflow.js tf.data.Dataset.take()用法及代码示例

Tensorflow.js是由Google开发的开源库，用于在浏览器或节点环境中运行机器学习模型以及深度学习神经网络。

.take() 方法用于生成最大的数据集数数所述数据集中的最重要的项目。

用法:

take(count)

Parameters:

count:用于创建不同数据集的是所述数据集中存在的元素数量。此外，如果计数未定义或为负数或高于指定数据集的维度，则新创建的数据集将保存给定数据集的每个项目。

返回值：它返回 tf.data.Dataset。

范例1：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Defining dataset formed of an array of
// numbers and calling take() method
const res = tf.data.array([11, 12, 31, 43, 15, 64]).take(4);
  
// Calling forEachAsync() method and
// Printing output
await res.forEachAsync(op => console.log(op));

输出：

范例2：

Javascript


// Importing the tensorflow.js library
import * as tf from "@tensorflow/tfjs"
  
// Calling forEachAsync(), take() method 
// and printing output
await tf.data.array([31.1, 81.2, 5.1, 0, NaN, 'a']).
take(10.7).forEachAsync(op => console.log(op));

输出：

31.1
81.2
5.1
0
NaN
a

参考： https://js.tensorflow.org/api/latest/#tf.data.Dataset.take

相关用法

注：本文由纯净天空筛选整理自nidhi1352singh大神的英文原创作品 Tensorflow.js tf.data.Dataset class .take() Method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。