Tensorflow.js tf.data.Dataset 类 .take() 方法

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

.take() 方法用于形成一个数据集，该数据集在所述数据集中具有最大数量的最重要的项目。

句法：

take(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));

输出：

11
12
31
43 

示例 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